onlytojay

/

conv_ae_mnist

like 0

• "torch.storage._load_from_bytes",
• "cloudpickle.cloudpickle._function_setstate",
• "torch.nn.modules.activation.ReLU",
• "pythae.models.base.base_utils.ModelOutput",
• "cloudpickle.cloudpickle._builtin_type",
• "torch.nn.modules.container.Sequential",
• "cloudpickle.cloudpickle._make_function",
• "torch.nn.modules.container.ModuleList",
• "torch._utils._rebuild_parameter",
• "torch._utils._rebuild_tensor_v2",
• "pythae.models.nn.base_architectures.BaseDecoder",
• "builtins.int",
• "collections.OrderedDict",
• "torch.nn.modules.batchnorm.BatchNorm2d",
• "cloudpickle.cloudpickle.subimport",
• "typing.List",
• "cloudpickle.cloudpickle._make_skeleton_class",
• "torch.nn.modules.conv.ConvTranspose2d",
• "torch.nn.modules.activation.Sigmoid",
• "cloudpickle.cloudpickle._class_setstate",
• "builtins.dict",
• "builtins.type",
• "_operator.getitem",
• "torch.nn.modules.linear.Linear",
• "torch.Tensor"

How to fix it?