fix: validation frequency

Former-commit-id: 99560cfe0c9d3ec2a7bc7571dc3a89365fa73d8c
javiribera · Mar 7, 2018 · d5264bc · d5264bc
1 parent 885a986
commit d5264bc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plant-locator/train.py b/plant-locator/train.py
@@ -64,7 +64,7 @@
                     help='only use the first N images of the training dataset')
 parser.add_argument('--max-valset-size', type=int, default=np.inf, metavar='N',
                     help='only use the first N images of the validation dataset')
-parser.add_argument('--val-freq', default=1, type=int, metavar='F',
+parser.add_argument('--val-freq', default=4, type=int, metavar='F',
                     help='Run validation after F epochs.')
 parser.add_argument('--env-name', default='Pure U-Net', type=str, metavar='NAME',
                     help='Name of the environment in Visdom')
@@ -271,7 +271,7 @@
         epoch += 1
         continue
 
-    if (epoch + 1) % args.val_frequency != 0:
+    if (epoch + 1) % args.val_freq != 0:
         epoch += 1
         continue