train error #3
Description
stack expects each tensor to be equal size, but got [3, 1632, 1224] at entry 0 and [3, 1296, 968] at entry 1
{{ message }}
stack expects each tensor to be equal size, but got [3, 1632, 1224] at entry 0 and [3, 1296, 968] at entry 1