shaneR:

However, this changes the size of the original inputs. Thus, there is an error when attemping to load_state_dict of the model.


Changing the input shape shouldn’t change the parameter shapes and the inputs would also not be stored in the model.state_dict(). It seems you’ve manipulated the model parameters instead.
You would have to apply these manipulations to the original model before loading the state_dict.