This commit is contained in:
Glenn Jocher 2020-02-09 09:12:45 -08:00
parent 8bc7648b38
commit 8bc9f56564
1 changed files with 1 additions and 1 deletions

View File

@ -264,7 +264,7 @@ def train():
# Multi-Scale training
if opt.multi_scale:
if ni / accumulate % 1 == 0: #  adjust (67% - 150%) every 10 batches
if ni / accumulate % 1 == 0: #  adjust img_size (67% - 150%) every 1 batch
img_size = random.randrange(img_sz_min, img_sz_max + 1) * 32
sf = img_size / max(imgs.shape[2:]) # scale factor
if sf != 1: