Upload 3 files
82fdd8b
verified
epoch_199.ckpt
Detected Pickle imports (61)
- "rrnco.envs.atsp.generator.ATSPGenerator",
- "rrnco.models.nn.attn_freenet.FeedForward",
- "omegaconf.listconfig.ListConfig",
- "collections.OrderedDict",
- "torchrl.data.tensor_specs.BoundedContinuous",
- "torch.bool",
- "rrnco.models.nn.attn_freenet.AFTFull",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.container.ModuleList",
- "rrnco.models.env_embeddings.atsp.ATSPInitEmbedding",
- "torch._C._nn.scaled_dot_product_attention",
- "collections.defaultdict",
- "rrnco.envs.atsp.sampler.Real_World_Sampler",
- "torch.LongStorage",
- "__builtin__.dict",
- "torch.nn.modules.activation.ReLU",
- "rrnco.models.decoder.RRNetDecoder",
- "torch._utils._rebuild_parameter",
- "omegaconf.base.ContainerMetadata",
- "__builtin__.list",
- "rrnco.models.env_embeddings.atsp.ContextualGating",
- "omegaconf.dictconfig.DictConfig",
- "typing.Any",
- "rrnco.models.nn.attn_freenet.DistAngleFusion",
- "torch.device",
- "rrnco.models.nn.attn_freenet.TransformerFFN",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.dropout.Dropout",
- "rrnco.envs.atsp.env.ATSPEnv",
- "rrnco.models.nn.attn_freenet.Normalization",
- "torchrl.data.tensor_specs.Composite",
- "torchrl.data.tensor_specs.ContinuousBox",
- "torch.nn.modules.container.Sequential",
- "torchrl.data.tensor_specs.UnboundedContinuous",
- "torch.nn.modules.activation.Sigmoid",
- "torchrl.data.tensor_specs.BoundedDiscrete",
- "rl4co.models.nn.env_embeddings.dynamic.StaticEmbedding",
- "torch.int64",
- "rrnco.models.decoder.RRNet_PointerAttention",
- "torch.Size",
- "torchrl.data.tensor_specs.UnboundedDiscrete",
- "rl4co.models.nn.mlp.MLP",
- "torch.nn.modules.linear.Identity",
- "torch.nn.modules.instancenorm.InstanceNorm1d",
- "rl4co.models.nn.env_embeddings.context.TSPContext",
- "rrnco.models.policy.RRNetPolicy",
- "rl4co.data.dataset.TensorDictDataset",
- "rrnco.models.encoder.RRNetEncoder",
- "rrnco.models.nn.attn_freenet.AttnFree_Block",
- "torch.nn.modules.container.ModuleDict",
- "omegaconf.base.Metadata",
- "rrnco.models.nn.attn_freenet.Attn_Free_Layer",
- "torch.nn.modules.linear.Linear",
- "torch.BoolStorage",
- "collections.Counter",
- "torch.float32",
- "__builtin__.set",
- "torch.FloatStorage",
- "rrnco.models.nn.attn_freenet.AttnFreeNet",
- "torch.ByteStorage",
- "__builtin__.long"
How to fix it?
41.2 MB
Upload 3 files