TCLResearchEurope

/

deit3_small_patch16_224_pruned_66

like 0

• "torch_dag.structured_modules.FlattenModule",
• "torch_dag.structured_modules.ParameterModule",
• "torch_dag.structured_modules.SliceModule",
• "torch_dag.structured_modules.AuxiliaryTokenModule",
• "torch.nn.modules.conv.Conv2d",
• "torch.nn.modules.dropout.Dropout",
• "torch.nn.modules.container.ModuleList",
• "torch.FloatStorage",
• "torch._utils._rebuild_parameter",
• "timm.models.vision_transformer.LayerScale",
• "torch_dag.structured_modules.AddModule",
• "torch.nn.modules.normalization.LayerNorm",
• "__builtin__.slice",
• "torch.nn.modules.linear.Identity",
• "collections.OrderedDict",
• "torch_dag.structured_modules.TfMatmulModule",
• "torch.nn.modules.activation.Softmax",
• "torch.nn.modules.linear.Linear",
• "torch_dag.structured_modules.TransposeModule",
• "__builtin__.set",
• "torch.nn.modules.activation.GELU",
• "torch_dag.structured_modules.ConcatModule",
• "torch._utils._rebuild_tensor_v2"

How to fix it?