dugoalberto

/

3D_U_Net

like 0

• "__builtin__.set",
• "torch.nn.modules.dropout.Dropout3d",
• "__main__.ImprovedUNet3D",
• "torch.nn.modules.instancenorm.InstanceNorm3d",
• "torch.nn.modules.activation.LeakyReLU",
• "torch.nn.parallel.data_parallel.DataParallel",
• "torch.FloatStorage",
• "torch.device",
• "torch._utils._rebuild_parameter",
• "collections.OrderedDict",
• "torch.nn.modules.conv.ConvTranspose3d",
• "torch._utils._rebuild_tensor_v2",
• "torch.nn.modules.container.Sequential",
• "torch.nn.modules.pooling.MaxPool3d",
• "torch.nn.modules.conv.Conv3d"

How to fix it?

• "collections.OrderedDict",
• "torch._utils._rebuild_parameter",
• "__main__.ImprovedUNet3D",
• "torch._utils._rebuild_tensor_v2",
• "torch.nn.modules.container.Sequential",
• "torch.FloatStorage",
• "torch.nn.modules.conv.ConvTranspose3d",
• "torch.nn.modules.dropout.Dropout3d",
• "torch.nn.modules.pooling.MaxPool3d",
• "torch.nn.modules.conv.Conv3d",
• "torch.nn.modules.instancenorm.InstanceNorm3d",
• "torch.nn.parallel.data_parallel.DataParallel",
• "torch.nn.modules.activation.LeakyReLU",
• "__builtin__.set",
• "torch.device"

How to fix it?