Skip to content

Commit

Permalink
minor changes
Browse files Browse the repository at this point in the history
  • Loading branch information
kocchop committed Oct 23, 2021
1 parent db4f16f commit 77fcf90
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
8 changes: 4 additions & 4 deletions train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down
2 changes: 1 addition & 1 deletion validate.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 77fcf90

Please sign in to comment.