修改 pytorch中的model zoo下载后的模型的保存目录
发布日期:2021-05-04 21:02:11 浏览次数:12 分类:技术文章

本文共 158 字,大约阅读时间需要 1 分钟。

网上看到有人暴力修改代码实现将下载的hub或model zoo的保存目录修改为自己喜欢的位置。

本质上pytorch已经提供环境变量重新定义模型下载后的保存目录:

 

import osos.environ["TORCH_HOME"]="E:\\pth2onnx\\models"   #你要的保存目录,修改环境变量即可

 

上一篇:pytorch中一些CV模型的预训练模型下载并转换为onnx模型
下一篇:torchvision 中的预训练模型及相关参数

发表评论

最新留言

做的很好,不错不错
[***.243.131.199]2025年03月22日 02时50分02秒