minor changes

Author: kocchop

train.py

@@ -72,11 +72,11 @@ def getOpt():
     parser.add_argument("--hr_height", type=int, default=192, help="dense depth height")
     parser.add_argument("--hr_width", type=int, default=256, help="dense depth width")
     parser.add_argument("--channels", type=int, default=1, help="depth image has only 1 channel")
-    parser.add_argument("--sample_interval", type=int, default=10, help="interval between saving image samples")
-    parser.add_argument("--validation_interval", type=int, default=12, help="interval between two consecutive validations")
-    parser.add_argument("--checkpoint_interval", type=int, default=15, help="batch interval between model checkpoints")
+    parser.add_argument("--sample_interval", type=int, default=4000, help="interval between saving image samples")
+    parser.add_argument("--validation_interval", type=int, default=4000, help="interval between two consecutive validations")
+    parser.add_argument("--checkpoint_interval", type=int, default=400, help="batch interval between model checkpoints")
     parser.add_argument("--residual_blocks", type=int, default=17, help="number of residual blocks in the generator")
-    parser.add_argument("--warmup_batches", type=int, default=5, help="number of batches with pixel-wise loss only")
+    parser.add_argument("--warmup_batches", type=int, default=250, help="number of batches with pixel-wise loss only")
     parser.add_argument("--lambda_adv", type=float, default=5e-3, help="adversarial loss weight")
     parser.add_argument("--lambda_pixel", type=float, default=1e-2, help="pixel-wise loss weight")
     parser.add_argument("--gpus", metavar='DEV_ID', default=None,

validate.py

@@ -62,7 +62,7 @@ def getOpt():
     parser.add_argument("--hr_width", type=int, default=256, help="high res. image width")
     parser.add_argument("--channels", type=int, default=1, help="number of image channels")
     parser.add_argument("--residual_blocks", type=int, default=17, help="number of residual blocks in the generator")
-    parser.add_argument("--validation_interval", type=int, default=12, help="interval between two consecutive validations")
+    parser.add_argument("--validation_interval", type=int, default=4000, help="interval between two consecutive validations")
     parser.add_argument("--lambda_adv", type=float, default=5e-3, help="adversarial loss weight")
     parser.add_argument("--lambda_pixel", type=float, default=1e-2, help="pixel-wise loss weight")
     parser.add_argument("--gpus", metavar='DEV_ID', default=None,