huggingface / huggingface/setfit
Error when uploading model checkpoints to Weights & Biases
- Dominant language
- Jupyter Notebook
- Stars
- 2.8k
- Forks
- 267
- Avg merge
- 36m
- Merged PRs (30d)
- 5
Description
When settting `os.environ["WANDB_LOG_MODEL"] = "end"` prior to the training loop and specify `report_to='wandb'` in `TrainingArguments`, I receive the following error:
```
Loading best SentenceTransformer model from step 14551.
Traceback (most recent call last):
File "", line 31, in main
trainer.train()
File "/usr/local/lib/python3.10/dist-packages/setfit/trainer.py", line 410, in train
self.train_embeddings(*full_parameters, args=args)
File "/usr/local/lib/python3.10/dist-packages/setfit/trainer.py", line 463, in train_embeddings
self._train_sentence_transformer(
File "/usr/local/lib/python3.10/dist-packages/setfit/trainer.py", line 687, in _train_sentence_transformer
self.control = self.callback_handler.on_train_end(args, self.state, self.control)
File "/usr/local/lib/python3.10/dist-packages/transformers/trainer_callback.py", line 366, in on_train_end
return self.call_event("on_train_end", args, state, control)
File "/usr/local/lib/python3.10/dist-packages/transformers/trainer_callback.py", line 407, in call_event
result = getattr(callback, event)(
File "/usr/local/lib/python3.10/dist-packages/transformers/integrations/integration_utils.py", line 771, in on_train_end
fake_trainer = Trainer(args=args, model=model, tokenizer=tokenizer)
File "/usr/local/lib/python3.10/dist-packages/transformers/trainer.py", line 337, in __init__
enable_full_determinism(self.args.seed) if self.args.full_determinism else set_seed(self.args.seed)
AttributeError: 'TrainingArguments' object has no attribute 'full_determinism'
```
Occurs only when I set the `WANDB_LOG_MODEL` environment variable, otherwise it runs smoothly.
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.