about train

Open yanghan0617 opened this issue 2 years ago • 0 comments

this is the problem.

RuntimeError: stack expects each tensor to be equal size, but got [3, 256, 256] at entry 0 and [3, 156, 256] at entry 4

Please tell me how to solve it.

Jan 19 '24 09:01 yanghan0617