Any pre-processing required?

[aditipanda]

I seem to be getting this error:

Am I missing any resizing step?

ValueError: Input and output must have the same number of spatial dimensions, but got input with spatial dimensions of [352] and output size of (256, 256). Please provide input tensor in (N, C, d1, d2, ...,dK) format and output size in (o1, o2, ...,oK) format.