nirmeshdell
/

distilbert-imdb-mlflow

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

distilbert-imdb-mlflow

1 contributor

History: 2 commits

nirmeshdell's picture

End of training

78b2e90 verified 2 months ago

.gitattributes

1.52 kB

initial commit 2 months ago
README.md

1.06 kB

End of training 2 months ago
config.json

587 Bytes

End of training 2 months ago
model.safetensors

263 MB
LFS

End of training 2 months ago
training_args.bin
Detected Pickle imports (9)
- "transformers.trainer_utils.IntervalStrategy",
- "transformers.training_args.TrainingArguments",
- "transformers.trainer_utils.SchedulerType",
- "transformers.trainer_utils.HubStrategy",
- "torch.device",
- "accelerate.utils.dataclasses.DistributedType",
- "transformers.training_args.OptimizerNames",
- "transformers.trainer_pt_utils.AcceleratorConfig",
- "accelerate.state.PartialState"
How to fix it?
4.98 kB
LFS

End of training 2 months ago