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
......@@ -753,6 +753,7 @@ class WandbCallback(TrainerCallback):
_watch_model = os.getenv("WANDB_WATCH", "false")
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.run._label(code="transformers_trainer")
def on_train_begin(self, args, state, control, model=None, **kwargs):
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