Update README.md
796107e - 1.52 kB initial commit
- 2.31 kB Update README.md
- 40.2 kB Upload 2 files
modules.pt Detected Pickle imports (19)
- "torch_dag.structured_modules.AddModule",
- "__builtin__.set",
- "torch_dag.structured_modules.ReshapeModule",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.linear.Linear",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.activation.Softmax",
- "torch.LongStorage",
- "torch.nn.modules.conv.Conv2d",
- "torch_dag.structured_modules.MulModule",
- "torch.FloatStorage",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.batchnorm.BatchNorm2d",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.modules.container.ModuleList",
- "collections.OrderedDict",
- "torch.nn.modules.activation.Sigmoid",
- "torch_dag.structured_modules.GlobalMeanPool2DModule",
- "torch_dag.structured_modules.Conv2DSameModule"
How to fix it?
28.5 MB Upload 2 files