Ubuntu
Ubuntu系统下相关技术
Amon O_*
这个作者很懒,什么都没留下…
展开
-
Pytorch保存、加载模型
1. 保存加载整个模型 torch.save(model,'model.pth') # save torch,load('model.pth') #load 2.仅加载参数 torch.save(model.state_dict(),'model.pth')#save model.model() #创建网络结构 params = torch.load('model.pth') #获取参数 model.load_state_dict(params) 3.仅加载部分参数 pretrained_dict原创 2020-05-22 19:42:34 · 154 阅读 · 0 评论 -
Ubuntu 16 安装虚拟环境
一. 安装虚拟环境 1.安装virtualenv和virtualenvwrapper sudo pip install virtualenv sudo pip install virtualenvwrapper原创 2019-08-06 18:56:47 · 119 阅读 · 0 评论