Update README.md
7acf6d2 - 1_Pooling Add SetFit model
- runs End of training
- 1.52 kB initial commit
- 13 Bytes End of training
- 13.3 kB Update README.md
- 1.78 kB Update config.json
- 116 Bytes Add SetFit model
model_head.pkl Detected Pickle imports (7)
- "torch._utils._rebuild_tensor_v2",
- "torch.device",
- "torch._utils._rebuild_parameter",
- "collections.OrderedDict",
- "torch.storage._load_from_bytes",
- "setfit.modeling.SetFitHead",
- "torch.nn.modules.linear.Linear"
How to fix it?
50.7 kB Add SetFit model - 349 Bytes Add SetFit model
- 438 MB End of training
- 53 Bytes Add SetFit model
- 280 Bytes Add SetFit model
- 711 kB End of training
- 405 Bytes End of training
training_args.bin Detected Pickle imports (8)
- "torch.device",
- "transformers.trainer_utils.SchedulerType",
- "transformers.training_args.TrainingArguments",
- "transformers.trainer_utils.HubStrategy",
- "accelerate.state.PartialState",
- "accelerate.utils.dataclasses.DistributedType",
- "transformers.training_args.OptimizerNames",
- "transformers.trainer_utils.IntervalStrategy"
How to fix it?
4.03 kB End of training - 232 kB Add SetFit model