Allow extra arguments (#625)

2019-08-21 20:34:31 -06:00 · 2019-08-21 20:34:31 -06:00 · ec9020e851
parent 0b33b536b7
commit ec9020e851
1 changed files with 1 additions and 1 deletions
--- a/mnist/model.py
+++ b/mnist/model.py
@ -62,7 +62,7 @@ def parse_arguments():
                      default=0.01,
                      help='Learning rate for training.')

-  args = parser.parse_args()
+  args = parser.parse_known_args()[0]
  return args