Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add epoch option (#933)

This commit is contained in:
baeseongsu 2020-02-25 23:46:01 +09:00 committed by GitHub
parent 4d36e76cbc
commit 932770771b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pl_examples/basic_examples/lightning_module_template.py
View File

@ -250,6 +250,7 @@ class LightningTemplateModel(pl.LightningModule):
parser.add_argument('--data_root', default=os.path.join(root_dir, 'mnist'), type=str) parser.add_argument('--data_root', default=os.path.join(root_dir, 'mnist'), type=str)
# training params (opt) # training params (opt)
parser.add_argument('--epochs', default=20, type=int)
parser.add_argument('--optimizer_name', default='adam', type=str) parser.add_argument('--optimizer_name', default='adam', type=str)
parser.add_argument('--batch_size', default=64, type=int) parser.add_argument('--batch_size', default=64, type=int)
return parser return parser