ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-04-08 06:34:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix example requirements

Browse Source
This commit is contained in:
allegroai 2021-10-03 02:16:26 +03:00
parent 56230faa5b
commit 3643446f56
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
examples/frameworks/ignite/requirements.txt Normal file
View File

@ -0,0 +1 @@
pytorch-ignite

3
examples/frameworks/pytorch-lightning/pytorch_lightning_example.py
View File

@ -61,7 +61,7 @@ if __name__ == '__main__':
parser = ArgumentParser() parser = ArgumentParser()
parser.add_argument('--batch_size', default=32, type=int) 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 = pl.Trainer.add_argparse_args(parser)
parser = LitClassifier.add_model_specific_args(parser) parser = LitClassifier.add_model_specific_args(parser)
args = parser.parse_args() args = parser.parse_args()
@ -86,7 +86,6 @@ if __name__ == '__main__':
# training # training
# ------------ # ------------
trainer = pl.Trainer.from_argparse_args(args) trainer = pl.Trainer.from_argparse_args(args)
trainer.max_epochs = args.epochs
trainer.fit(model, train_loader, val_loader) trainer.fit(model, train_loader, val_loader)
# ------------ # ------------

2
examples/frameworks/pytorch-lightning/requirements.txt
View File

@ -1,4 +1,4 @@
clearml clearml
pytorch_lightning ~= 1.1.2 pytorch_lightning >= 1.1.2
torch torch
torchvision torchvision