Fetching metadata from the HF Docker repository...
Upload best_model.pt
f613bf1
verified
-
1.57 kB
Upload 5 files
-
679 Bytes
Update Dockerfile
-
195 Bytes
initial commit
-
163 kB
Upload best_model.pt
-
7.53 kB
Update main.py
-
41.7 MB
Upload 5 files
-
135 Bytes
Update requirements.txt
traffic_transformer_model_2.pth
Detected Pickle imports (17)
- "torch.nn.modules.activation.MultiheadAttention",
- "torch.nn.modules.linear.Linear",
- "torch._utils._rebuild_tensor_v2",
- "torch._C._nn.gelu",
- "torch.nn.modules.activation.ReLU",
- "torch.nn.modules.container.ModuleList",
- "torch.nn.modules.transformer.TransformerEncoderLayer",
- "collections.OrderedDict",
- "__main__.TransformerForecaster",
- "torch.FloatStorage",
- "torch.nn.modules.transformer.TransformerEncoder",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "torch.nn.modules.dropout.Dropout",
- "__builtin__.set",
- "torch.nn.modules.normalization.LayerNorm",
- "torch._utils._rebuild_parameter"
How to fix it?
172 kB
Upload 5 files