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
......@@ -713,6 +713,7 @@ class WandbCallback(TrainerCallback):
hp_search = state.is_hyper_param_search
if hp_search:
self._wandb.finish()
self._initialized = False
if not self._initialized:
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