image size attribut not working about yolor

[WongKinYiu]

input size should be 128x for p6 models.

[abdelhak51]

thank you @WongKinYiu I solved the problem so I found that --img-size accepts 64 , 320 only

[0chandansharma]

So, this problem is coming because of Resize of test and train images

These functions are responsible for changing the size, For TESTING please change the shape according to these lines of code..............

# Image sizes
gs = 64 #int(max(model.stride))  # grid size (max stride)
imgsz, imgsz_test = [check_img_size(x, gs) for x in opt.img_size]

def check_img_size(img_size, s=32):
    # Verify img_size is a multiple of stride s
    new_size = make_divisible(img_size, int(s))  # ceil gs-multiple
    if new_size != img_size:
        print('WARNING: --img-size %g must be multiple of max stride %g, updating to %g' % (img_size, s, new_size))
    return new_size

def make_divisible(x, divisor):
    # Returns x evenly divisible by divisor
    return math.ceil(x / divisor) * divisor