PyTorch 自带很多预训练模型,在使用时会自动下载,本文记录修改下载位置的方法。
C:\Users\<username>\.cache
,很可能占用 C 盘几个 G 的空间,尝试修改该路径pretrained
为True时,PyTorch会调用torch.utils
的load_state_dict_from_url
函数load_state_dict_from_url
函数最终调用torch.utils.model_zoo.load_url
函数model_dir
就是保存的目录,这里它默认会使用环境变量TORCH_HOME
TORCH_HOME
的值为~/.cache
%USERPROFILE%\.cache
,其中%USERPROFILE%
表示环境变量%USERPROFILE%
的值,它对应Linux的~
。import os
os.environ['TORCH_HOME']='E:/Data/torch-cache'
2022-08-25 14:53:36,183 - timm.models.helpers - INFO - Loading pretrained weights from url (https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/wide_resnet50_racm-8234f177.pth)
Downloading: "https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/wide_resnet50_racm-8234f177.pth" to E:/Data/torch-cache\hub\checkpoints\wide_resnet50_racm-8234f177.pth