FidelityWM

/

planning-baselines

Like 0

Follow

fidelity-world-model 2

• "torch._tensor._rebuild_from_type_v2",
• "collections.OrderedDict",
• "torch.FloatStorage",
• "torch.Tensor",
• "torch._utils._rebuild_tensor_v2"

How to fix it?

• "accelerate.utils.dataclasses.TorchDynamoPlugin",
• "torch.nn.modules.container.ModuleList",
• "torch.FloatStorage",
• "models.vit.Transformer",
• "models.vit.FeedForward",
• "torch.optim.adamw.AdamW",
• "models.vqvae.ResBlock",
• "torch.nn.modules.conv.Conv2d",
• "torch.nn.modules.activation.GELU",
• "models.vit.Attention",
• "accelerate.utils.dataclasses.DynamoBackend",
• "__builtin__.set",
• "models.vqvae.Quantize",
• "torch.nn.modules.activation.Softmax",
• "torch._utils._rebuild_parameter",
• "torch.nn.modules.normalization.LayerNorm",
• "torch.nn.modules.activation.ReLU",
• "torch._utils._rebuild_tensor_v2",
• "models.vqvae.Decoder",
• "torch.nn.modules.container.Sequential",
• "torch.nn.modules.conv.Conv1d",
• "__builtin__.dict",
• "collections.OrderedDict",
• "accelerate.utils.dataclasses.DistributedType",
• "torch.nn.modules.conv.ConvTranspose2d",
• "torch.optim.adam.Adam",
• "torch.device",
• "models.proprio.ProprioceptiveEmbedding",
• "accelerate.state.GradientState",
• "collections.defaultdict",
• "accelerate.optimizer.AcceleratedOptimizer",
• "torch.nn.modules.linear.Linear",
• "models.vit.ViTPredictor",
• "accelerate.state.AcceleratorState",
• "models.vqvae.VQVAE",
• "torch.nn.modules.dropout.Dropout"

How to fix it?