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)的主要内容,如果未能解决你的问题,请参考以下文章
AutoTokenizer.from_pretrained 无法加载本地保存的预训练标记器 (PyTorch)