ppak10
/

defect-classification-distilbert-baseline-15-epochs

Generated from Trainer

Model card Files Files and versions Community

defect-classification-distilbert-baseline-15-epochs

Ctrl+K

Ctrl+K

1 contributor

History: 33 commits

ppak10's picture

Trained model with classification head weights

bc6b0b2 verified 3 months ago

.gitattributes

1.52 kB

initial commit 3 months ago
README.md

2.46 kB

Trained model with classification head weights 3 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
LFS

Training in progress, epoch 15 3 months ago
model.safetensors

265 MB
LFS

Training in progress, epoch 15 3 months ago
special_tokens_map.json

125 Bytes

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

711 kB

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

1.23 kB

Training in progress, epoch 1 3 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
LFS

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

232 kB

Training in progress, epoch 1 3 months ago