保存位置

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
Logo
智源数据社区

更多推荐

cover

自然语言处理(NLP)-下游任务&数据集：语言模型、机器翻译、问答、文本分类、情感分析、文本生成、自动摘要、命名实体识别、阅读理解、自然语言推理、信息提取、词性标注、共指消解、实体链接【＞200项】

avatar 智源数据社区

利用科大讯飞开放平台 进行自然语言处理（NLP）Python

最近在做聊天机器人的人工智能实践，需要用到依存句法分析和语义依存分析，所以利用强大的中文语言技术平台注册及快速入门网址 https://www.xfyun.cn/快速入门文档 https://www.xfyun.cn/doc/platform/quickguide.htmlIP白名单设置运行demo时，会出现类似{"code":"10105","data":{},"desc":"ill...

avatar 智源数据社区

Coursera自然语言处理专项课程04：Natural Language Processing with Attention Models笔记 Week02

Text Summarization using Transformer

avatar 智源数据社区