Add config back to WandB init

ebsmothers · ebsmothers · commit 23addf034df2 · 2021-11-01T18:35:19.000Z
diff --git a/mmf/utils/logger.py b/mmf/utils/logger.py
@@ -425,7 +425,7 @@ def __init__(
             )
 
         self._wandb = wandb
-        self._wandb_init = dict(entity=entity, project=project)
+        self._wandb_init = dict(entity=entity, config=config, project=project)
         wandb_kwargs = deepcopy(config.training.wandb)
         wandb_kwargs.pop("enabled")
         wandb_kwargs.pop("entity")

Original file line number	Diff line number	Diff line change
`@@ -425,7 +425,7 @@ def __init__(`
`425`	`425`	`)`
`426`	`426`
`427`	`427`	`self._wandb = wandb`
`428`		`- self._wandb_init = dict(entity=entity, project=project)`
	`428`	`+ self._wandb_init = dict(entity=entity, config=config, project=project)`
`429`	`429`	`wandb_kwargs = deepcopy(config.training.wandb)`
`430`	`430`	`wandb_kwargs.pop("enabled")`
`431`	`431`	`wandb_kwargs.pop("entity")`