akshatmehta98
/

distilbert-imdb-mlflow

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

distilbert-imdb-mlflow

1 contributor

History: 30 commits

akshatmehta98's picture

Training in progress, epoch 1

abff403 verified 7 months ago

.gitattributes

1.52 kB

initial commit 7 months ago
README.md

1.08 kB

End of training 7 months ago
config.json

636 Bytes

Training in progress, epoch 1 7 months ago
model.safetensors
263 MB
LFS

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

Training in progress, epoch 1 7 months ago