From 55442d8bb13042faed3e152ec607e8d286913016 Mon Sep 17 00:00:00 2001 From: Teng Li Date: Wed, 2 Dec 2020 17:20:34 +0800 Subject: [PATCH] fix type conflict issue --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 3215aa2..ba5d852 100644 --- a/main.py +++ b/main.py @@ -20,7 +20,7 @@ parser.add_argument('--batch_size', dest='batch_size', type=int, default=16, help='number of samples in one batch') parser.add_argument('--patch_size', dest='patch_size', type=int, default=48, help='patch size') parser.add_argument('--start_lr', dest='start_lr', type=float, default=0.001, help='initial learning rate for adam') -parser.add_argument('--eval_every_epoch', dest='eval_every_epoch', default=20, help='evaluating and saving checkpoints every # epoch') +parser.add_argument('--eval_every_epoch', dest='eval_every_epoch', type=int, default=20, help='evaluating and saving checkpoints every # epoch') parser.add_argument('--checkpoint_dir', dest='ckpt_dir', default='./checkpoint', help='directory for checkpoints') parser.add_argument('--sample_dir', dest='sample_dir', default='./sample', help='directory for evaluating outputs')