Update main.py

mpc001 · Oct 9, 2020 · 599654a · 599654a
1 parent 2c57b00
commit 599654a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vae/main.py b/vae/main.py
@@ -14,7 +14,7 @@
 parser.add_argument('--epochs', type=int, default=10, metavar='N',
                     help='number of epochs to train (default: 10)')
 parser.add_argument('--no-cuda', action='store_true', default=False,
-                    help='enables CUDA training')
+                    help='disables CUDA training')
 parser.add_argument('--seed', type=int, default=1, metavar='S',
                     help='random seed (default: 1)')
 parser.add_argument('--log-interval', type=int, default=10, metavar='N',