From 4e640531daf6e93d95160d2cc3601858d61656f5 Mon Sep 17 00:00:00 2001 From: Brainway Date: Tue, 1 Nov 2022 02:44:12 +0000 Subject: [PATCH] update train.py. --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index dff4905..0da476a 100755 --- a/train.py +++ b/train.py @@ -292,7 +292,7 @@ def main(): parser.add_argument("--pretrained_model", type=str, default=None, help="load pretrained model") #None parser.add_argument("--output_dir", default="./output", type=str, help="The output directory where checkpoints will be written.") - parser.add_argument("--img_size", default=320, type=int, help="Resolution size") + parser.add_argument("--img_size", default=600, type=int, help="Resolution size") parser.add_argument("--train_batch_size", default=8, type=int, help="Total batch size for training.") parser.add_argument("--eval_batch_size", default=8, type=int,