pytorch加载默认模型位置

这是一只小菜鸡

于 2020-05-12 16:58:37 发布

阅读量2.3k

点赞数

文章标签： python

版权

在PyTorch中加载预训练模型通常会默认从特定文件夹读取。要改变模型加载路径，可以使用os模块来操作环境变量，从而指定模型所在的目录。这样在加载模型时，将会从设定的新路径读取。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在使用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