Fix an argument instruction in mnist_hogwild

jacobswan1 · Oct 25, 2017 · 9a02f2a · 9a02f2a
1 parent 7d0d413
commit 9a02f2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mnist_hogwild/main.py b/mnist_hogwild/main.py
@@ -14,7 +14,7 @@
 parser.add_argument('--test-batch-size', type=int, default=1000, metavar='N',
                     help='input batch size for testing (default: 1000)')
 parser.add_argument('--epochs', type=int, default=10, metavar='N',
-                    help='number of epochs to train (default: 2)')
+                    help='number of epochs to train (default: 10)')
 parser.add_argument('--lr', type=float, default=0.01, metavar='LR',
                     help='learning rate (default: 0.01)')
 parser.add_argument('--momentum', type=float, default=0.5, metavar='M',