RuntimeError: stack expects each tensor to be equal size, but got [3, 128, 128] at entry 0 and [3, 218, 178] at entry 15
RuntimeError: stack expects each tensor to be equal size, but got [3, 128, 128] at entry 0 and [3, 218, 178] at entry 15