mlreco.models.layers.gnn.message_passing.dgtransformer module¶
-
class
mlreco.models.layers.gnn.message_passing.dgtransformer.DeepGraphTransformer(cfg)[source]¶ Bases:
torch.nn.modules.module.Module-
__init__(cfg)[source]¶ Initializes internal Module state, shared by both nn.Module and ScriptModule.
-
__module__= 'mlreco.models.layers.gnn.message_passing.dgtransformer'¶
-
training: bool¶
-