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')

你可能感兴趣的:(随笔,python,深度学习,开发语言)