Upload train.py
a487597
-
1.38 kB
initial commit
-
41.7 kB
Update README.md
-
5.77 kB
Upload . with huggingface_hub
-
27.1 kB
Upload . with huggingface_hub
model.pkl
Detected Pickle imports (26)
- "transformers.models.distilbert.modeling_distilbert.Embeddings",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.distilbert.modeling_distilbert.DistilBertModel",
- "torch.FloatStorage",
- "transformers.models.distilbert.modeling_distilbert.FFN",
- "transformers.models.distilbert.modeling_distilbert.DistilBertForSequenceClassification",
- "torch.nn.modules.dropout.Dropout",
- "torch.LongStorage",
- "torch.nn.modules.sparse.Embedding",
- "torch._C._nn.gelu",
- "transformers.models.distilbert.modeling_distilbert.Transformer",
- "transformers.models.distilbert.configuration_distilbert.DistilBertConfig",
- "__main__.BertModule",
- "transformers.activations.GELUActivation",
- "__builtin__.dict",
- "torch.nn.modules.linear.Linear",
- "collections.OrderedDict",
- "torch.optim.adamw.AdamW",
- "__builtin__.set",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch._utils._rebuild_parameter",
- "transformers.models.distilbert.modeling_distilbert.TransformerBlock",
- "torch.nn.modules.container.ModuleList",
- "transformers.models.distilbert.modeling_distilbert.MultiHeadSelfAttention",
- "collections.defaultdict",
- "torch._utils._rebuild_tensor_v2"
How to fix it?
1.61 GB
Upload . with huggingface_hub
-
5.64 kB
Upload train.py