From 3643446f56382848e24ab0da844af84236ddb11f Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Sun, 3 Oct 2021 02:16:26 +0300 Subject: [PATCH] Fix example requirements --- examples/frameworks/ignite/requirements.txt | 1 + .../frameworks/pytorch-lightning/pytorch_lightning_example.py | 3 +-- examples/frameworks/pytorch-lightning/requirements.txt | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) create mode 100644 examples/frameworks/ignite/requirements.txt diff --git a/examples/frameworks/ignite/requirements.txt b/examples/frameworks/ignite/requirements.txt new file mode 100644 index 00000000..febdc0b7 --- /dev/null +++ b/examples/frameworks/ignite/requirements.txt @@ -0,0 +1 @@ +pytorch-ignite diff --git a/examples/frameworks/pytorch-lightning/pytorch_lightning_example.py b/examples/frameworks/pytorch-lightning/pytorch_lightning_example.py index fa78ec5a..362585e3 100644 --- a/examples/frameworks/pytorch-lightning/pytorch_lightning_example.py +++ b/examples/frameworks/pytorch-lightning/pytorch_lightning_example.py @@ -61,7 +61,7 @@ if __name__ == '__main__': parser = ArgumentParser() parser.add_argument('--batch_size', default=32, type=int) - parser.add_argument('--epochs', default=3, type=int) + parser.add_argument('--max_epochs', default=3, type=int) parser = pl.Trainer.add_argparse_args(parser) parser = LitClassifier.add_model_specific_args(parser) args = parser.parse_args() @@ -86,7 +86,6 @@ if __name__ == '__main__': # training # ------------ trainer = pl.Trainer.from_argparse_args(args) - trainer.max_epochs = args.epochs trainer.fit(model, train_loader, val_loader) # ------------ diff --git a/examples/frameworks/pytorch-lightning/requirements.txt b/examples/frameworks/pytorch-lightning/requirements.txt index cd407d0e..9fed39e4 100644 --- a/examples/frameworks/pytorch-lightning/requirements.txt +++ b/examples/frameworks/pytorch-lightning/requirements.txt @@ -1,4 +1,4 @@ clearml -pytorch_lightning ~= 1.1.2 +pytorch_lightning >= 1.1.2 torch torchvision