SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix(wandb): fix config (#9489)

Browse Source
This commit is contained in:
Boris Dayma
2021-01-08 20:32:02 +01:00
committed by GitHub
parent 02e05fb0a5
commit 1c19b423bf
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/integrations.py
View File

@@ -333,9 +333,9 @@ class WandbCallback(TrainerCallback):
logger.warning( logger.warning(
"W&B installed but not logged in. Run `wandb login` or set the WANDB_API_KEY env variable." "W&B installed but not logged in. Run `wandb login` or set the WANDB_API_KEY env variable."
) )
self._wandb = wandb
else:
self._wandb = None self._wandb = None
else:
self._wandb = wandb
self._initialized = False self._initialized = False
def setup(self, args, state, model, reinit, **kwargs): def setup(self, args, state, model, reinit, **kwargs):