From 2ccf68cf96416faca899e4af8be911ac56271be9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 5 Nov 2018 09:08:48 +0100 Subject: [PATCH] add multi_scale train option to argparser --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 4d87934e..cb9eb39d 100644 --- a/train.py +++ b/train.py @@ -12,7 +12,7 @@ parser.add_argument('-data_config_path', type=str, default='cfg/coco.data', help parser.add_argument('-cfg', type=str, default='cfg/yolov3.cfg', help='cfg file path') parser.add_argument('-img_size', type=int, default=32 * 13, help='size of each image dimension') parser.add_argument('-resume', default=False, help='resume training flag') -parser.add_argument('-multi_scale', default=True, help='train at random img_size 320-608') +parser.add_argument('-multi_scale', default=True, help='train at random img_size 320-608') # ensure memory for 608 size opt = parser.parse_args() print(opt)