pytorch读取预训练模型并修改模型下载位置
下载模型import torchvisionmodel = torchvision.models.vgg13_bn(pretrained=True)print(model)模型下载的位置默认在C盘，这个可以根据环境变量来设置。首先找到并打开 “查看高级系统设置” ，win10可以直接在下面的搜索导航中搜索到。打开之后进入环境变量点击‘新建’然后输入变量名 TORCH_HOME 和你所需要的储存位置
·
- 下载模型
import torchvision
model = torchvision.models.vgg13_bn(pretrained=True)
print(model)
- 模型下载的位置默认在C盘，这个可以根据环境变量来设置。
首先找到并打开 “查看高级系统设置” ，win10可以直接在下面的搜索导航中搜索到。
- 打开之后进入环境变量
- 点击‘新建’
- 然后输入变量名 TORCH_HOME 和你所需要的储存位置，之后确定保存好。
- 我使用的jupyter notebook, 所以需要重启jupyter notebook，pycharm,vscode等其他编辑器请自行尝试是否需要重启。
更多推荐
15
0- 0
已为社区贡献1条内容
所有评论(0)