说明
使用torchvision.model加载预训练好的模型时,发现默认下载路径在系统盘下面的用户目录下(这个你执行的时候就会发现),即C:\用户名\.cache\torch\.checkpoints
下,便于统一管理,我决定修改model的存放路径,在网上找了很久都没有很好的解决方法,只能自己尝试,现将解决方案给出,供大家参考~
操作环境
- windows10 + Anaconda
- torch:1.1.0
- torchvision:0.3.0
加载方式
以加载vgg16为例,首先定义网络结构
class CNN(nn.Module):
def __init__(self, usegpu=