PyTorch下载的预训练模型的保存位置(Windows)

Posted Xavier Jiezou

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PyTorch下载的预训练模型的保存位置(Windows)相关的知识,希望对你有一定的参考价值。

保存位置

C:\\Users\\xxx\\.cache\\torch\\hub\\checkpoints\\

xxx替换为你的用户名。

项目场景

迁移学习的时候一般需要用到预训练模型,那么预训练模型的保存位置是在哪呢?

示例代码

from torchvision import models
model = models.vgg19(pretrained=True)

这里以VGG19为例,只要把pretrained参数设置为True就会下载模型,而模型的保存位置是在C盘用户目录下的\\.cache\\torch\\hub\\checkpoints\\

Downloading: "https://download.pytorch.org/models/vgg19-dcbb9e9d.pth" to C:\\Users\\ghgxj\\.cache\\torch\\hub\\checkpoints\\vgg19-dcbb9e9d.pth

温馨提示

关于PyTorch中迁移学习的教程可以看我的这篇文章。

以上是关于PyTorch下载的预训练模型的保存位置(Windows)的主要内容,如果未能解决你的问题,请参考以下文章

Pytorch 中的预训练模型

AutoTokenizer.from_pretrained 无法加载本地保存的预训练标记器 (PyTorch)

Pytorch - 跳过计算每个时期的预训练模型的特征

如果我们扩展或减少同一模型的层,我们仍然可以从 Pytorch 中的预训练模型进行训练吗?

为啥新层在修改后的预训练 pytorch 模型中被忽略?

PyTorch 的预训练,是时候学习一下了