Create README.md
6ad7194
verified
-
1.52 kB
initial commit
-
2.5 kB
Create README.md
-
3.17 kB
Create inference.py
model_85_nn_.pth
Detected Pickle imports (19)
- "torch.FloatStorage",
- "torchvision.models.efficientnet.MBConv",
- "torchvision.ops.misc.Conv2dNormActivation",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.batchnorm.BatchNorm2d",
- "torch.nn.modules.activation.Sigmoid",
- "torch._utils._rebuild_parameter",
- "torchvision.models.efficientnet.EfficientNet",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.conv.Conv2d",
- "torch.nn.modules.dropout.Dropout",
- "torch._utils._rebuild_tensor_v2",
- "torchvision.ops.stochastic_depth.StochasticDepth",
- "torchvision.ops.misc.SqueezeExcitation",
- "torch.nn.modules.pooling.AdaptiveAvgPool2d",
- "torch.nn.modules.activation.SiLU",
- "torch.LongStorage",
- "collections.OrderedDict",
- "__builtin__.set"
How to fix it?
71.1 MB
85% Accuracy of face shape classifcation