RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 608 and 416 in dimension 2 at /pytorch/aten/src/TH/generic/THTensor.cpp:689

[Naveennavik]

May i know clear what to do to get rid of this error? @andy-yun @mickolka @carry-xz

[Ginbor]

you can switch off the "get_different_scale" function in dataset.py line 63 OR you can set the the input image size (608, 416 ... ) before starting a new epoch, line 203 of train.py.
for example put the folowing text instead 198 and 199 lines in train.py:
init_width = (random.randint(0,9) + 10)*32
init_height = init_width.

bad solution, but works