ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 09:07:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix example

Browse Source
This commit is contained in:
allegroai 2021-10-03 03:25:04 +03:00
parent 39d8d8f0cb
commit b0a3166a93
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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()