slow-eagle
/

ELL861

Model card Files Files and versions

ELL861 / kaggle /working

8 MB

Ctrl+K

Ctrl+K

1 contributor

History: 2 commits

slow-eagle's picture

Final trained model checkpoint

74bf0f0 verified 9 months ago

full_model.pth
Detected Pickle imports (14)
- "__main__.MHA",
- "collections.OrderedDict",
- "torch.nn.modules.sparse.Embedding",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.activation.GELU",
- "__main__.FFN",
- "torch._utils._rebuild_parameter",
- "__main__.DecoderTransformer",
- "torch.nn.modules.container.ModuleList",
- "torch.FloatStorage",
- "torch.nn.modules.linear.Linear",
- "__main__.TransformerBlock",
- "torch._utils._rebuild_tensor_v2",
- "__builtin__.set"
How to fix it?
8 MB
xet

Final trained model checkpoint 9 months ago