The problem of torch loading custom models

-----------------------------------------------------------------------------------------------------------------------------

General loading method reference ：pytorch- Model saving and loading - The sea of stars in the shell - Blog Garden pytorch- Model saving and loading Loading model parameters and selection are determined by the saved model data structure , Therefore, we must first determine the method and data structure of saving the model Save the model # Model weight parameters model.state_dict() 'https://www.cnblogs.com/tian777/p/14924518.html