vincent-doan
Minor change
2ee69f4
-
2.75 kB
Minor change
srgan_checkpoint.pth
Detected Pickle imports (14)
- "torch._utils._rebuild_tensor_v2",
- "torch._utils._rebuild_parameter",
- "__main__.ResidualBlock",
- "torch.nn.modules.pixelshuffle.PixelShuffle",
- "torch.nn.modules.activation.PReLU",
- "torch.LongStorage",
- "torch.nn.modules.conv.Conv2d",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.batchnorm.BatchNorm2d",
- "collections.OrderedDict",
- "__main__.GeneratorResnet",
- "__builtin__.set",
- "torch.nn.modules.activation.Tanh",
- "torch.FloatStorage"
How to fix it?
6.34 MB
add SRGAN