huggingface / huggingface/setfit

Error when uploading model checkpoints to Weights & Biases

Open
#464 3 comments 0 reactions 0 assignees View on GitHub
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.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.