当前位置:   article > 正文

【PyTorch】下载的预训练模型的保存位置(Windows)_pretrained=true 下载的模型在哪

pretrained=true 下载的模型在哪

保存位置

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

xxx替换为你的用户名。

项目场景

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

示例代码

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

这里以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
  • 1

温馨提示

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/木道寻08/article/detail/772669?site
推荐阅读
相关标签
  

闽ICP备14008679号