Skip to content

GitLab

Unverified Commit 122b2657 authored by Bharat Ramanathan's avatar Bharat Ramanathan Committed by GitHub
Browse files

feat: add trainer label to wandb run upon initialization (#26466)

parent 4fdf47cd
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/integrations/integration_utils.py
View file @ 122b2657
...@@ -753,6 +753,7 @@ class WandbCallback(TrainerCallback): ...@@ -753,6 +753,7 @@ class WandbCallback(TrainerCallback):
_watch_model = os.getenv("WANDB_WATCH", "false") _watch_model = os.getenv("WANDB_WATCH", "false")
if not is_torch_tpu_available() and _watch_model in ("all", "parameters", "gradients"): if not is_torch_tpu_available() and _watch_model in ("all", "parameters", "gradients"):
self._wandb.watch(model, log=_watch_model, log_freq=max(100, state.logging_steps)) self._wandb.watch(model, log=_watch_model, log_freq=max(100, state.logging_steps))
self._wandb.run._label(code="transformers_trainer")
def on_train_begin(self, args, state, control, model=None, **kwargs): def on_train_begin(self, args, state, control, model=None, **kwargs):
if self._wandb is None: if self._wandb is None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment