diff --git a/grok/training.py b/grok/training.py
index 43b43df..3ffe804 100755
--- a/grok/training.py
+++ b/grok/training.py
@@ -47,7 +47,7 @@ def __init__(self, hparams: Namespace) -> None:
                         self.add_model_specific_args().
         """
         super().__init__()
-        self.hparams = hparams  # type: ignore
+        self.hparams.update(vars(hparams))  # type: ignore
         self.prepare_data()
 
         self.transformer = Transformer(