From d83f6bb9c9360dda1ef639af3af23ae73949d395 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 28 Jun 2019 13:53:00 -0400 Subject: [PATCH] changed read me --- examples/new_project_templates/lightning_module_template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py index 152bf5a9b5..9968a12de5 100644 --- a/examples/new_project_templates/lightning_module_template.py +++ b/examples/new_project_templates/lightning_module_template.py @@ -148,7 +148,7 @@ class LightningTemplateModel(LightningModule): return whatever optimizers we want here :return: list of optimizers """ - optimizer = optim.Adam(lr=self.hparams.learning_rate) + optimizer = optim.Adam(self.parameters(), lr=self.hparams.learning_rate) return [optimizer] def __dataloader(self, train):