保存和加载整个模型以及参数 不需要再加载模型了,参数和模型都保存下来了 # torch.save(model, 'mymodel.pkl') # model1 = torch.load('mymodel.pkl') 已有模型加载已保存的参数 需要在再次创建模型, torch.save(model.state_dict(), 'params.pkl') model1=Cnn() model1.load_state_dict(torch.load('params.pkl'))