FCHD-Fully-Convolutional-Head-Detector icon indicating copy to clipboard operation
FCHD-Fully-Convolutional-Head-Detector copied to clipboard

Published 20 hours ago • verified publisher icon aditya-vora

why can't I change the invalid batch_size

Open KelvinHuang666 opened this issue 5 years ago • 2 comments

when I change the invalid batch_size=128 or other number !=1, RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 861 and 811 in dimension 3 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1307

KelvinHuang666 avatar May 27 '19 10:05 KelvinHuang666

when I change the invalid batch_size=128 or other number !=1, RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 861 and 811 in dimension 3 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1307

I've met the same problem, have you already solved it?

LeonardNiu avatar Jul 18 '19 13:07 LeonardNiu

yes, i encountered the same issue. after i pad the label bbox to the same size, the error has gone, but i finally got the following error below, it's raised from trainer.py. raise ValueError('Currently only batch size 1 is supported.')

i believe this project is mainly for demo purpose where batch size= 1 works fine, i guess i just have to wait longer for the training process.

BTW, brilliant work with visdom, i cannot believe it's realtime and interactive!

DANNALI35 avatar Aug 22 '19 07:08 DANNALI35