sleepy boy

Lightning-AI · Oct 14, 2020 · 1fff9d4 · 1fff9d4
1 parent ecf1b48
commit 1fff9d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/core/saving.py b/pytorch_lightning/core/saving.py
@@ -382,7 +382,7 @@ def save_hparams_to_yaml(config_yaml, hparams: Union[dict, Namespace]) -> None:
         except TypeError as err:
             warn(f"Skipping '{k}' parameter because it is not possible to safely dump to YAML.")
         else:
-            hparams[k] = v
+            hparams_allowed[k] = v
 
     # saving the standard way
     with fs.open(config_yaml, "w", newline="") as fp: