import torch
model = torch.hub.list('pytorch/vision')
Run Code Online (Sandbox Code Playgroud)
我的pytorch版本是1.0.0,但是无法加载集线器,这是为什么?
您将需要torch >= 1.1.0使用torch.hub属性。
或者,尝试下载此 hub.py文件,然后尝试以下代码:
import hub
model = hub.list('pytorch/vision', force_reload=False)
Run Code Online (Sandbox Code Playgroud)
论据:
github:repo_owner/repo_name[:tag_name]必需,具有带有可选标记/分支的格式的字符串。master如果未指定,则为默认分支。例子:pytorch/vision[:hub]
force_reload:可选,是否放弃现有缓存并强制重新下载。默认为False.