【发布时间】:2019-07-31 20:54:10
【问题描述】:
import torch
model = torch.hub.list('pytorch/vision')
我的pytorch版本是1.0.0,但是无法加载hub,这是为什么?
【问题讨论】:
标签: pytorch
import torch
model = torch.hub.list('pytorch/vision')
我的pytorch版本是1.0.0,但是无法加载hub,这是为什么?
【问题讨论】:
标签: pytorch
您需要torch >= 1.1.0 才能使用torch.hub 属性。
或者,尝试下载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。
【讨论】: