Python PyTorch list用法及代码示例

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

用法:

torch.hub.list(github, force_reload=False, skip_validation=False)

参数：

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

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

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

返回：

可用的可调用入口点

返回类型：

列表

列出 github 指定的 repo 中可用的所有可调用入口点。

示例

>>> entrypoints = torch.hub.list('pytorch/vision', force_reload=True)