pytorch加载默认模型位置

在使用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/squeezenet1_1-f364aa15.pth" to /media/sunkj/WorkAndStudy/pytorch/models/checkpoints/squeezenet1_1-f364aa15.pth
100%|██████████| 4.74M/4.74M [00:01<00:00, 2.57MB/s]