Fetching metadata from the HF Docker repository...
Nikhil Mudhalwadkar
adding model
77b7934
epoch=9-step=17780.ckpt
Detected Pickle imports (18)
- "__builtin__.set",
- "torch.nn.modules.conv.Conv2d",
- "collections.OrderedDict",
- "torch.LongStorage",
- "torch.nn.modules.container.Sequential",
- "app.discriminator.patch_gan.Discriminator",
- "app.discriminator.patch_gan.CNNBlock",
- "app.generator.unetGen.Generator",
- "torch.nn.modules.activation.LeakyReLU",
- "torch.nn.modules.conv.ConvTranspose2d",
- "torch.FloatStorage",
- "torch.nn.modules.activation.Tanh",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.activation.ReLU",
- "app.generator.unetGen.Block",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.modules.batchnorm.BatchNorm2d"
How to fix it?
686 MB
adding model