Python PyTorch help用法及代码示例

本文简要介绍python语言中 torch.hub.help 的用法。

用法:

torch.hub.help(github, model, force_reload=False, skip_validation=False)

参数：

• github(string) -格式为 的字符串，带有可选标签/分支。如果未指定 tag_name ，则假定默认分支为 main (如果存在)，否则为 master 。示例：‘pytorch/vision:0.10’

• model(string) -在 repo 的 hubconf.py 中定义的入口点名称字符串

• force_reload(bool,可选的) -是否丢弃现有缓存并强制重新下载。默认为 False 。

• skip_validation(bool,可选的) -如果 False ，torchhub 将检查 github 参数指定的分支或提交是否正确属于存储库所有者。这将向 GitHub API 发出请求；您可以通过设置 GITHUB_TOKEN 环境变量来指定非默认 GitHub 令牌。默认为 False 。

显示入口点 model 的文档字符串。

示例

>>> print(torch.hub.help('pytorch/vision', 'resnet18', force_reload=True))