diff --git a/train.py b/train.py index f963fdf1..ca99daca 100644 --- a/train.py +++ b/train.py @@ -64,7 +64,7 @@ def train(hyp): imgsz_min, imgsz_max, imgsz_test = opt.img_size # img sizes (min, max, test) # Image Sizes - gs = 64 # (pixels) grid size + gs = 32 # (pixels) grid size assert math.fmod(imgsz_min, gs) == 0, '--img-size %g must be a %g-multiple' % (imgsz_min, gs) opt.multi_scale |= imgsz_min != imgsz_max # multi if different (min, max) if opt.multi_scale: