在调用torchvision.models
下面的预训练模型的时候,好比resnet18
会直接开始从重新给你下载这个模型,下载位置是在当前用户文件夹的.cache
文件夹下,这样会给下次使用造成使用的困扰,为了改变下载的位置,方便之后的使用,对其进行修改。
查看resnet18
的定义位置,可以看到
调用的是load_state_dict_from_url
,该函数来自于utils
,查看可以看到很简单
调用的是torch.hub
,转到定义查看这个函数,看到的是
未修改的时候这里是model_dir=None
,将其修改为你要存储的位置就可以。