Apply suggestions from code review

Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com>
Lightning-AI · Oct 15, 2020 · 75a2134 · 75a2134
1 parent acb0fb4
commit 75a2134
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pytorch_lightning/core/saving.py b/pytorch_lightning/core/saving.py
@@ -381,6 +381,7 @@ def save_hparams_to_yaml(config_yaml, hparams: Union[dict, Namespace]) -> None:
             yaml.dump(v)
         except TypeError as err:
             warn(f"Skipping '{k}' parameter because it is not possible to safely dump to YAML.")
+            hparams[k] = type(v).__name__
         else:
             hparams_allowed[k] = v