diff --git a/train.py b/train.py index 7094ca9..e471c56 100644 --- a/train.py +++ b/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, diff --git a/validate.py b/validate.py index d6264b8..f35457c 100644 --- a/validate.py +++ b/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,