torch.hub.load 加载本地模型（已解决）

最新推荐文章于 2024-09-12 14:17:53 发布

娃要大红花

最新推荐文章于 2024-09-12 14:17:53 发布

阅读量1.1w

点赞数 12

分类专栏：随笔文章标签： python 深度学习开发语言

本文链接：https://blog.csdn.net/michael_quan2013/article/details/129833357

版权

随笔专栏收录该内容

6 篇文章 0 订阅

订阅专栏

背景

运行网上的项目，有时会卡住或者超时，原因是 torch.hub.load 默认会去网上找模型，有时会因为网络问题而报错

解决方法

不让 torch.hub.load 联网下载模型，改为 torch.hub.load 加载本地模型。本地模型默认的下载路径是：/root/.cache/torch/hub……

原始：vad_model, funcs = torch.hub.load(repo_or_dir="snakers4/silero-vad", model="silero_vad", trust_repo=True)

改为：vad_model, funcs = torch.hub.load(repo_or_dir="/root/.cache/torch/hub/snakers4_silero-vad_master/", model="silero_vad", trust_repo=True, source='local')