Use save_hyperparameters instead of 'self.hparams = hparams'

HHousen · May 11, 2021 · e36e033 · e36e033
1 parent b2d81af
commit e36e033
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/abstractive.py b/src/abstractive.py
@@ -90,7 +90,7 @@ class AbstractiveSummarizer(pl.LightningModule):
     def __init__(self, hparams):
         super(AbstractiveSummarizer, self).__init__()
 
-        self.hparams = hparams
+        self.save_hyperparameters(hparams)
 
         if len(self.hparams.dataset) <= 1:
             self.hparams.dataset = self.hparams.dataset[0]

diff --git a/src/extractive.py b/src/extractive.py
@@ -106,7 +106,7 @@ def __init__(self, hparams, embedding_model_config=None, classifier_obj=None):
         hparams.tokenizer_no_use_fast = getattr(hparams, "tokenizer_no_use_fast", False)
         hparams.data_type = getattr(hparams, "data_type", "none")
 
-        self.hparams = hparams
+        self.save_hyperparameters(hparams)
         self.forward_modify_inputs_callback = None
 
         if not embedding_model_config: