updates
This commit is contained in:
parent
8bc7648b38
commit
8bc9f56564
2
train.py
2
train.py
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue