From e2eb8324613605f921bd951e439f2f2bb54a18ca Mon Sep 17 00:00:00 2001 From: thomas chaton Date: Wed, 15 Sep 2021 08:01:41 +0100 Subject: [PATCH] [bugfix] Resolve example after LightningCLI update (#9520) --- pl_examples/basic_examples/profiler_example.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pl_examples/basic_examples/profiler_example.py b/pl_examples/basic_examples/profiler_example.py index 598287ea64..cec8a9727a 100644 --- a/pl_examples/basic_examples/profiler_example.py +++ b/pl_examples/basic_examples/profiler_example.py @@ -35,6 +35,7 @@ from pytorch_lightning.profiler.pytorch import PyTorchProfiler from pytorch_lightning.utilities.cli import LightningCLI DEFAULT_CMD_LINE = ( + "fit", "--trainer.max_epochs=1", "--trainer.limit_train_batches=15", "--trainer.limit_val_batches=15",