From 932770771b4891a588da3b09f7e4ec719124db77 Mon Sep 17 00:00:00 2001 From: baeseongsu Date: Tue, 25 Feb 2020 23:46:01 +0900 Subject: [PATCH] add epoch option (#933) --- pl_examples/basic_examples/lightning_module_template.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pl_examples/basic_examples/lightning_module_template.py b/pl_examples/basic_examples/lightning_module_template.py index fabe1850..edc482c7 100644 --- a/pl_examples/basic_examples/lightning_module_template.py +++ b/pl_examples/basic_examples/lightning_module_template.py @@ -250,6 +250,7 @@ class LightningTemplateModel(pl.LightningModule): parser.add_argument('--data_root', default=os.path.join(root_dir, 'mnist'), type=str) # training params (opt) + parser.add_argument('--epochs', default=20, type=int) parser.add_argument('--optimizer_name', default='adam', type=str) parser.add_argument('--batch_size', default=64, type=int) return parser