ppak10
/

defect-classification-distilbert-baseline-05-epochs

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

defect-classification-distilbert-baseline-05-epochs

1 contributor

History: 7 commits

ppak10's picture

Trained model with classification head weights

2f936d3 verified 25 days ago

.gitattributes

1.52 kB

initial commit 25 days ago
README.md

1.82 kB

Trained model with classification head weights 25 days 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 5 25 days ago
model.safetensors

265 MB
LFS

Training in progress, epoch 5 25 days ago
special_tokens_map.json

125 Bytes

Training in progress, epoch 1 25 days ago
tokenizer.json

711 kB

Training in progress, epoch 1 25 days ago
tokenizer_config.json

1.23 kB

Training in progress, epoch 1 25 days 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 25 days ago
vocab.txt

232 kB

Training in progress, epoch 1 25 days ago