about train #137
Description
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.
{{ message }}
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.