Revert "fix lightning warning"

This reverts commit 7aff4eab1e.
This commit is contained in:
Kashif Rasul
2022-03-31 14:44:29 +02:00
parent 7aff4eab1e
commit 007db12b02
4 changed files with 6 additions and 6 deletions
+1 -1
View File
@@ -221,7 +221,7 @@
" weight_decay: float = 1e-8,\n",
" ) -> None:\n",
" super().__init__()\n",
" self.save_hyperparameters(ignore=['loss', 'model'])\n",
" self.save_hyperparameters()\n",
" self.model = model\n",
" self.loss = loss\n",
" self.lr = lr\n",
+1 -1
View File
@@ -15,7 +15,7 @@ class TFTLightningModule(pl.LightningModule):
weight_decay: float = 1e-8,
) -> None:
super().__init__()
self.save_hyperparameters(ignore=["loss", "model"])
self.save_hyperparameters()
self.model = model
self.loss = loss
self.lr = lr
+1 -1
View File
@@ -15,7 +15,7 @@ class TransformerLightningModule(pl.LightningModule):
weight_decay: float = 1e-8,
) -> None:
super().__init__()
self.save_hyperparameters(ignore=["loss", "model"])
self.save_hyperparameters()
self.model = model
self.loss = loss
self.lr = lr
File diff suppressed because one or more lines are too long