在使用pytorch之时,免不了要加载预习训练好的默认模型,如使用models.squeezenet1_1
,此时会默认将模型加载在~/.torch/models/
文件夹下,可以使用对环境变量的操作将其路径改变,以加载到指定文件夹。
对于模型的默认路径,是通过环境变量TORCH_HOME
进行操控。通过os控制命令行修改环境变量。
os.environ['TORCH_HOME'] = '/media/sunkj/WorkAndStudy/pytorch/models/'
再进行模型加载,即可。结果如下:
resnet34 = models.squeezenet1_1(pretrained=True, num_classes=1000)
Downloading: "https://download.pytorch.org/models/s