torch.hub.load 加载本地模型(已解决)

背景

运行网上的项目,有时会卡住或者超时,原因是 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')

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2023年7月15日
下一篇 2023年7月15日

相关推荐