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

Update mnist_trainer.py

parent 10f4e41b
No related branches found
No related tags found
No related merge requests found
......@@ -20,7 +20,7 @@ def main(hparams):
if __name__ == '__main__':
parser = ArgumentParser()
parser = ArgumentParser(add_help=False)
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 register or to comment