ppak10
/

defect-classification-distilbert-baseline-20-epochs

Generated from Trainer

Model card Files Files and versions

defect-classification-distilbert-baseline-20-epochs

266 MB

1 contributor

History: 22 commits

ppak10's picture

Trained model with classification head weights

31523cf verified 11 months ago

.gitattributes

1.52 kB

initial commit 11 months ago
README.md

2.77 kB

Trained model with classification head weights 11 months ago
classification_head.pt
Detected Pickle imports (3)
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "torch.FloatStorage"
What is a pickle import?
13.9 kB
xet

Training in progress, epoch 20 11 months ago
model.safetensors

265 MB
xet

Training in progress, epoch 20 11 months ago
special_tokens_map.json

125 Bytes

Training in progress, epoch 1 11 months ago
tokenizer.json

711 kB

Training in progress, epoch 1 11 months ago
tokenizer_config.json

1.23 kB

Training in progress, epoch 1 11 months ago
training_args.bin
Detected Pickle imports (10)
- "transformers.trainer_utils.IntervalStrategy",
- "transformers.trainer_utils.HubStrategy",
- "transformers.trainer_pt_utils.AcceleratorConfig",
- "accelerate.state.PartialState",
- "transformers.trainer_utils.SaveStrategy",
- "accelerate.utils.dataclasses.DistributedType",
- "transformers.training_args.TrainingArguments",
- "transformers.trainer_utils.SchedulerType",
- "transformers.training_args.OptimizerNames",
- "torch.device"
How to fix it?
5.37 kB
xet

Training in progress, epoch 1 11 months ago
vocab.txt

232 kB

Training in progress, epoch 1 11 months ago