Pytorch预训练模型下载并加载(以VGG为例)自定义路径

简述 通常来说,Pytorch用torchvision调用vgg之类的模型话,若是电脑在cache(Pytorch硬编码的一个地址)(若是在环境变量中添加了TORCH_HOME 和TORCH_MODEL_ZOO的话,就是在这两个位置的联合的路径下,好比TORCH_MODEL_ZOO\model)不然就是在TORCH_HOME\models或者是~/.torch/modelsgit 好比,个人就是C
相关文章
相关标签/搜索