Update minimum stride to 32 (#2266)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-02-21 21:46:42 -08:00
1 parent 095d2c11d8
commit e27ca0d845
2 files changed
+4 -3

No files matched your search

+1 -1
View File
@@ -161,7 +161,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
del ckpt, state_dict
# Image sizes
gs = int(model.stride.max()) # grid size (max stride)
gs = max(int(model.stride.max()), 32) # grid size (max stride)
nl = model.model[-1].nl # number of detection layers (used for scaling hyp['obj'])
imgsz, imgsz_test = [check_img_size(x, gs) for x in opt.img_size] # verify imgsz are gs-multiples