用yolov5训练了一个权重,项目只能部署在本地
torch.hub.load(repo_or_dir, model, *args, source='github', force_reload=False, verbose=True, skip_validation=False, **kwargs)
从 github 存储库或本地目录加载模型。
注意:加载模型是典型的用例,但这也可用于加载其他对象,例如分词器、损失函数等。
如果 <span><span>source</span></span>
是”github”, <span><span>repo_or_dir</span></span>
则应 <span><span>repo_owner/repo_name[:tag_name]</span></span>
采用带有可选标签/分支的形式。
如果 <span><span>source</span></span>
是”local “, <span><span>repo_or_dir</span></span>
则应为本地目录的路径。
- repo_or_dir ( string ) – 如果
<span><span>source</span></span>
是 ‘github’,这应该对应于<span><span>repo_owner/repo_name[:tag_name]</span></span>
具有可选标签/分支格式的 github 存储库,例如 ‘pytorch/vision:0.10’。如果<span><span>tag_name</span></span>
未指定,则假定默认分支为<span><span>main</span></span>
存在,否则为<span><span>master</span></span>
。如果<span><span>source</span></span>
是”local”,则它应该是本地目录的路径。 - model ( string ) – 在 repo/dir’s 中定义的可调用(入口点)的名称
<span><span>hubconf.py</span></span>
。 - *args( 可选)– callable 的相应参数
<span><span>model</span></span>
。 - source ( string , optional ) – ‘github’ 或 ‘local’。指定如何
<span><span>repo_or_dir</span></span>
解释。默认为”github”。 - force_reload ( bool , optional ) – 是否无条件强制重新下载github repo。如果 没有任何影响source = ‘local’。默认为
<span><span>False</span></span>
- verbose ( bool , optional ) – 如果
<span><span>False</span></span>
,静音有关命中本地缓存的消息。请注意,有关首次下载的消息无法静音。如果<span><span>source</span></span> <span><span>=</span></span> <span><span>'local'</span></span>
没有任何影响。默认为<span><span>True</span></span>
。 - skip_validation ( bool , optional ) – 如果
<span><span>False</span></span>
,torchhub 将检查<span><span>github</span></span>
参数指定的分支或提交是否正确属于 repo 所有者。这将向 GitHub API 发出请求;您可以通过设置<span><span>GITHUB_TOKEN</span></span>
环境变量来指定非默认 GitHub 令牌 。默认为<span><span>False</span></span>
。 - **kwargs (可选) – callable 的相应 kwargs
<span><span>model</span></span>
。
<span><span>model</span></span>
使用给定 <span><span>*args</span></span>
和调用时可调用 的输出 <span><span>**kwargs</span></span>
。
Original: https://blog.csdn.net/qq_34317565/article/details/122299172
Author: 陆三斤
Title: torch.hub.load()解析,如何加载本地权重
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/706083/
转载文章受原作者版权保护。转载请注明原作者出处!