From 4414d1285656fc3c59205c1e66dea5e8672f5b46 Mon Sep 17 00:00:00 2001
From: Lauro Langosco di Langosco <langosco.lauro@gmail.com>
Date: Fri, 14 Jan 2022 14:25:38 +0100
Subject: [PATCH] Avoid AttributeError resulting from Pytorch Lightning update

---
 grok/training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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(