This commit is contained in:
Glenn Jocher 2020-02-07 10:53:09 -08:00
parent 145ea67a2e
commit 106b1961b6
1 changed files with 1 additions and 1 deletions

View File

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