Up default number of training epochs

iver56 · Jun 30, 2019 · 2c60838 · 2c60838
1 parent 77fad65
commit 2c60838
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tcvc/train.py b/tcvc/train.py
@@ -45,7 +45,7 @@
         "--testBatchSize", type=int, default=1, help="testing batch size"
     )
     parser.add_argument(
-        "--nEpochs", type=int, default=32, help="number of epochs to train for"
+        "--nEpochs", type=int, default=50, help="number of epochs to train for"
     )
     parser.add_argument("--input_nc", type=int, default=1, help="input image channels")
     parser.add_argument(