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):
trainer.fit(model)
if __name__ == '__init__':
if __name__ == '__main__':
parser = ArgumentParser()
parser.add_argument('--gpus', type=str, default=None)
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 to comment