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,