wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-11 12:31:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix max_epochs setup in basic example (#1105)

Browse Source 
tested only for the CPU version
This commit is contained in:
Ondrej Platek
2020-03-12 12:42:52 -04:00
committed by GitHub
parent fb27a771f8
commit 1383f64a5f
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pl_examples/basic_examples/cpu_template.py
+1 -1
View File
@@ -28,7 +28,7 @@ def main(hparams):
# ------------------------
# 2 INIT TRAINER
# ------------------------
trainer = pl.Trainer()
trainer = pl.Trainer(max_epochs=hparams.epochs)
# ------------------------
# 3 START TRAINING
pl_examples/basic_examples/gpu_template.py
+1
View File
@@ -29,6 +29,7 @@ def main(hparams):
# 2 INIT TRAINER
# ------------------------
trainer = pl.Trainer(
max_epochs=hparams.epochs,
gpus=hparams.gpus,
distributed_backend=hparams.distributed_backend,
use_amp=hparams.use_16bit