Skip to content

GitLab

Unverified Commit c3d958b2 authored by Josh Tanner's avatar Josh Tanner Committed by GitHub
Browse files

reinitialize wandb config for each hyperparameter search run (#11945)

parent 99dbbdb9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/integrations.py
View file @ c3d958b2
...@@ -713,6 +713,7 @@ class WandbCallback(TrainerCallback): ...@@ -713,6 +713,7 @@ class WandbCallback(TrainerCallback):
hp_search = state.is_hyper_param_search hp_search = state.is_hyper_param_search
if hp_search: if hp_search:
self._wandb.finish() self._wandb.finish()
self._initialized = False
if not self._initialized: if not self._initialized:
self.setup(args, state, model, **kwargs) self.setup(args, state, model, **kwargs)
......
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