torch.hub.load()解析,如何加载本地权重

用yolov5训练了一个权重,项目只能部署在本地
官方文档

torch.hub.load(repo_or_dir, model, *args, source='github', force_reload=False, verbose=True, skip_validation=False, **kwargs)

[来源]
从 github 存储库或本地目录加载模型。
注意:加载模型是典型的用例,但这也可用于加载其他对象,例如分词器、损失函数等。
如果source是“github”,repo_or_dir则应repo_owner/repo_name[:tag_name]采用带有可选标签/分支的形式。
如果source是“local”,repo_or_dir则应为本地目录的路径。
参数
repo_or_dir ( string ) – 如果source是 'github',这应该对应于repo_owner/repo_name[:tag_name]具有可选标签/分支格式的 github 存储库,例如 'pytorch/vis

torch.hub.load()解析,如何加载本地权重最先出现在Python成神之路

版权声明:
作者:感冒的梵高
链接:https://www.techfm.club/p/12762.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>