wandb logging fix

Lightning-AI · Jul 2, 2020 · e2614bf · e2614bf
1 parent 1a40963
commit e2614bf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pytorch_lightning/loggers/wandb.py b/pytorch_lightning/loggers/wandb.py
@@ -125,6 +125,7 @@ def watch(self, model: nn.Module, log: str = 'gradients', log_freq: int = 100):
     @rank_zero_only
     def log_hyperparams(self, params: Union[Dict[str, Any], Namespace]) -> None:
         params = self._convert_params(params)
+        params = self._flatten_dict(params)
         self.experiment.config.update(params, allow_val_change=True)
 
     @rank_zero_only