Haha, sorry for the confusion.
By “initially” I meant you should initialize the model in the same way as you have initialized it before storing the state_dict.
So, if you’ve created the model via model_ft, input_size = initialize_model(...), and stored the state_dict afterwards, you should use the same approach to recreate your model and then just load the state_dict again.