From 10f4e41b89289d2c6e0f500f34efa6be49e345d1 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Fri, 23 Aug 2019 14:32:55 -0400
Subject: [PATCH] Update mnist.py
---
research_seed/mnist/mnist.py | 2 --
1 file changed, 2 deletions(-)
diff --git a/research_seed/mnist/mnist.py b/research_seed/mnist/mnist.py
index 9803737..c48ea84 100644
--- a/research_seed/mnist/mnist.py
+++ b/research_seed/mnist/mnist.py
@@ -72,8 +72,6 @@ class CoolSystem(pl.LightningModule):
# training specific (for this model)
parser.add_argument('--max_nb_epochs', default=2, type=int)
- parser.add_argument('--gpus', type=str, default=None)
- parser.add_argument('--nodes', type=int, default=1)
return parser
--
GitLab