torchvison预训练模型位置修改

下载与加载位置固定为合适的位置

Posted by 周自横 on December 24, 2020

在调用torchvision.models下面的预训练模型的时候,好比resnet18会直接开始从重新给你下载这个模型,下载位置是在当前用户文件夹的.cache文件夹下,这样会给下次使用造成使用的困扰,为了改变下载的位置,方便之后的使用,对其进行修改。

查看resnet18的定义位置,可以看到

image-20201231152811858

调用的是load_state_dict_from_url,该函数来自于utils,查看可以看到很简单

image-20201231152947452

调用的是torch.hub,转到定义查看这个函数,看到的是image-20201231153111285

未修改的时候这里是model_dir=None,将其修改为你要存储的位置就可以。