Tensor size mismatch during prediction

[kcsumesh]

I received tensor mismatch error when making a prediction.

inputs = inputs.to(device = device)
outputs1, outputs2, outputs3 = model(inputs)

[long123524]

I received tensor mismatch error when making a prediction.

inputs = inputs.to(device = device) outputs1, outputs2, outputs3 = model(inputs)

Training and testing should be keep with the same input sizes. Herein, you seem to need the input size is 512x512.