add more models
54a7472
epoch-75.pt
Detected Pickle imports (19)
- "torch._utils._rebuild_parameter",
- "nets.graph_encoder.GraphAttentionEncoder",
- "torch.nn.modules.batchnorm.BatchNorm1d",
- "torch.nn.modules.container.Sequential",
- "torch.LongStorage",
- "torch.ByteStorage",
- "torch.nn.modules.linear.Linear",
- "__builtin__.set",
- "torch._utils._rebuild_tensor_v2",
- "torch.FloatStorage",
- "problems.vrp.problem_vrp.VRPDataset",
- "nets.graph_encoder.SkipConnection",
- "problems.vrp.problem_vrp.CVRP",
- "nets.graph_encoder.MultiHeadAttentionLayer",
- "nets.attention_model.AttentionModel",
- "nets.graph_encoder.Normalization",
- "torch.nn.modules.activation.ReLU",
- "nets.graph_encoder.MultiHeadAttention",
- "collections.OrderedDict"
How to fix it?
21.5 MB
add more models