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

[bugfix] Resolve example after LightningCLI update (#9520)

This commit is contained in:
thomas chaton 2021-09-15 08:01:41 +01:00 committed by GitHub
parent 20ebb5ccc4
commit e2eb832461
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/profiler_example.py
View File

@ -35,6 +35,7 @@ from pytorch_lightning.profiler.pytorch import PyTorchProfiler
from pytorch_lightning.utilities.cli import LightningCLI from pytorch_lightning.utilities.cli import LightningCLI
DEFAULT_CMD_LINE = ( DEFAULT_CMD_LINE = (
"fit",
"--trainer.max_epochs=1", "--trainer.max_epochs=1",
"--trainer.limit_train_batches=15", "--trainer.limit_train_batches=15",
"--trainer.limit_val_batches=15", "--trainer.limit_val_batches=15",