From b0a3166a934d29321f23f766023aa66f50f61f81 Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Sun, 3 Oct 2021 03:25:04 +0300 Subject: [PATCH] Fix example --- .../frameworks/pytorch-lightning/pytorch_lightning_example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/frameworks/pytorch-lightning/pytorch_lightning_example.py b/examples/frameworks/pytorch-lightning/pytorch_lightning_example.py index 362585e3..b65e5462 100644 --- a/examples/frameworks/pytorch-lightning/pytorch_lightning_example.py +++ b/examples/frameworks/pytorch-lightning/pytorch_lightning_example.py @@ -61,8 +61,8 @@ if __name__ == '__main__': parser = ArgumentParser() parser.add_argument('--batch_size', default=32, type=int) - parser.add_argument('--max_epochs', default=3, type=int) parser = pl.Trainer.add_argparse_args(parser) + parser.set_defaults(max_epochs=3) parser = LitClassifier.add_model_specific_args(parser) args = parser.parse_args()