From b07c24ff1d0f99928de722a8ed6c2f9df642f196 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Fri, 23 Aug 2019 14:30:32 -0400
Subject: [PATCH] Update mnist_trainer.py
---
research_seed/mnist/mnist_trainer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/research_seed/mnist/mnist_trainer.py b/research_seed/mnist/mnist_trainer.py
index 5dbd9db..a2f2b19 100644
--- a/research_seed/mnist/mnist_trainer.py
+++ b/research_seed/mnist/mnist_trainer.py
@@ -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)
--
GitLab