I printed out the test images shape and it’s like this :
intpus: torch.Size([50, 3, 1020, 2040]) scores: torch.Size([50, 1]).  For the whole testing data batches
input shape: torch.Size([1, 3, 1020, 2040])  For the data parallel input to one GPU.
It’s passed in the same model, and supposed to be [32, 128, 256])], but the size comes out before regressor [32, 127, 255] which doesn’t make sense to me…
I wil check with what you suggested “without data parallel”!
At the same time, I’m wondering whether it has something to do with me running the training model using Data parallel, and storing the .pt model in GPU directly with save_state()?
How are we supposed to load data parallel model? Is it different in any tense?