Sharukesh

/

attention-unet

like 1

• "torch.nn.modules.activation.Sigmoid",
• "__main__.ConvBlock",
• "__main__.AttentionGate",
• "torch.FloatStorage",
• "__main__.AttentionUNet",
• "torch.nn.modules.conv.Conv2d",
• "torch.nn.modules.batchnorm.BatchNorm2d",
• "torch.LongStorage",
• "torch.nn.modules.container.Sequential",
• "torch._utils._rebuild_tensor_v2",
• "torch.nn.modules.pooling.MaxPool2d",
• "torch.nn.modules.conv.ConvTranspose2d",
• "__builtin__.set",
• "collections.OrderedDict",
• "torch._utils._rebuild_parameter"

How to fix it?