donut icon indicating copy to clipboard operation
donut copied to clipboard

Published 20 hours ago • verified publisher icon clovaai

Error while training

Open rachiteagles opened this issue 1 year ago • 2 comments

I was training the model on a different dataset and image size i used was 250*250. I am getting error when i start training

RuntimeError: The size of tensor a (64) must match the size of tensor b (100) at non-singleton dimension 3

image

rachiteagles avatar Mar 30 '23 18:03 rachiteagles

Hi ! I am getting the same error, did you happen to figure out the problem ?

lauraminkova avatar Aug 03 '23 08:08 lauraminkova

Me as well, having the same issue when using image_size of 256 by 256. However, using a different image_size for the same dataset (specifying reconstruction on the feature extractor and encoder config), e.g. 480 by 360 works fine.

bikashg avatar Dec 21 '23 12:12 bikashg