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