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)