torch.hub.load 加载本地模型(已解决)
短信预约 -IT技能 免费直播动态提醒
背景
运行网上的项目,有时会卡住或者超时,原因是 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')
来源地址:https://blog.csdn.net/michael_quan2013/article/details/129833357
免责声明:
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341