torch.hub.set_custom_protocol('https') AttributeError: module 'torch.hub' has no attribute 'set_custom_protocol'
时间: 2024-04-27 15:21:55 浏览: 170
HTTP 协议自定义接口
这个问题可能是由于您使用的PyTorch版本较旧所导致的。在较旧的PyTorch版本中,`torch.hub`模块不支持`set_custom_protocol`方法。
解决方法是更新您的PyTorch版本。您可以使用以下命令更新到最新版本:
```
pip install torch --upgrade
```
如果您仍然无法更新到最新版本,或者您需要使用旧版本的PyTorch,您可以尝试在下载模型之前手动设置自定义协议,例如:
```python
import torch
import urllib.request
# 设置自定义协议
opener = urllib.request.build_opener()
opener.addheaders = [('User-agent', 'Mozilla/5.0')]
urllib.request.install_opener(opener)
torch.hub.download_url_to_file('https://github.com/argman/EAST/releases/download/1.0/east_vgg16.pth', 'east_vgg16.pth')
```
这个方法使用了Python的`urllib`库来手动设置HTTP请求头,从而避免使用`set_custom_protocol`方法。然后,您可以使用`torch.load`函数加载下载的模型参数。
阅读全文