Yeah I have tried that. All of them are corrupted. They are predicting constant points pretty much.
I have an implementation of the same network using a square input image instead of rectangular. That network is fine. I am getting the same errors when loading a previously trained model. That’s why I was wondering if there if something has changed internally when loading a trained model using state_dict function. This is the command I use to load a trained model for inference: net.load_state_dict(torch.load(trained_model_path)).
If everything seems okay, I think it might be a major bug in the code. Will look at that. Thank you for your help!