While running evaluation with batch size greater than one, the following error is thrown. > Error in model.eval(): stack expects each tensor to be equal size, but got [3, 427, 640] at entry 0 and [3, 230, 352] at entry 1