AttributeError:模块“ torch”没有属性“ hub”

时间:2019-06-26 03:38:05

标签: pytorch

import torch
model = torch.hub.list('pytorch/vision')

我的pytorch版本是1.0.0,但是我无法加载集线器,为什么?

1 个答案:

答案 0 :(得分:1)

尝试通过下载this hub.py文件,然后尝试以下代码:

import hub
model = hub.list('pytorch/vision', force_reload=False)

参数:

github:是必需的,格式为repo_owner/repo_name[:tag_name]的字符串,带有可选的标记/分支。如果未指定,则默认分支为master。 示例:pytorch/vision[:hub]

force_reload:可选,是否放弃现有缓存并强制重新下载。 默认值为False