Error tensor size during the training

LJOVO / TranSalNet

TranSalNet: Towards perceptually relevant visual saliency prediction. Neurocomputing (2022)

https://doi.org/10.1016/j.neucom.2022.04.080

MIT License

51 stars 10 forks source link

Error tensor size during the training #6

Closed fadhfaiz closed 1 month ago

fadhfaiz commented 2 months ago

Hi,

I'm encountering an error during training that says "The size of tensor a (300) must match the size of tensor b (108) at non-singleton dimension 1". Can you help me troubleshoot this error?

LJOVO commented 2 months ago

Hi, I think this error could be caused by an unexpected input size. You may resolve this error by resizing the input image to 384×288.