weixin_43855930的博客

1. 保存加载整个模型torch.save（model,'model.pth'） # savetorch,load（'model.pth'） #load2.仅加载参数torch.save(model.state_dict(),'model.pth')#savemodel.model() #创建网络结构params = torch.load('model.pth') #获取参数model.load_state_dict(params)3.仅加载部分参数pretrained_dict

一. 安装虚拟环境１．安装virtualenv和virtualenvwrappersudo pip install virtualenvsudo pip install virtualenvwrapper