Skip to content
Snippets Groups Projects
Unverified Commit b07c24ff authored by William Falcon's avatar William Falcon Committed by GitHub
Browse files

Update mnist_trainer.py

parent 220c646a
No related branches found
No related tags found
No related merge requests found
...@@ -19,7 +19,7 @@ def main(hparams): ...@@ -19,7 +19,7 @@ def main(hparams):
trainer.fit(model) trainer.fit(model)
if __name__ == '__init__': if __name__ == '__main__':
parser = ArgumentParser() parser = ArgumentParser()
parser.add_argument('--gpus', type=str, default=None) parser.add_argument('--gpus', type=str, default=None)
parser.add_argument('--nodes', type=int, default=1) parser.add_argument('--nodes', type=int, default=1)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment