diff --git a/research_seed/mnist/mnist.py b/research_seed/mnist/mnist.py index 9803737cd4408debf45b714ec0a4d3a7d9742fa4..c48ea849cfefb6e143e9bba6a769ec5ad27a9db9 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