Can not load model

[chenyaofo]

Can not load model

[chenyaofo]

There is a solution, I think it can help you.

import torch
import torch.hub as hub


def patch_hub_with_proxy():
    download_url_to_file = hub.download_url_to_file

    def _proxy_download_url_to_file(
        url: str,
        *args,
        **kwargs
    ):
        if url.startswith("https://github.com"):
            cdn_url = "https://ghproxy.com/" + url
            return download_url_to_file(cdn_url, *args, **kwargs)
    hub.download_url_to_file = _proxy_download_url_to_file

    def _git_archive_link(repo_owner, repo_name, ref):
        return f"https://github.com/{repo_owner}/{repo_name}/archive/refs/heads/{ref}.zip"
    hub._git_archive_link = _git_archive_link


patch_hub_with_proxy()

net = torch.hub.load("chenyaofo/pytorch-cifar-models", "cifar10_resnet20", force_reload=True, pretrained=True)

print(net)

[chenyaofo]

This maybe an network issue. I test the above code just now and it works well.

Thanks for your feedbacks. I an going to add some proxy tricks to make it can be accessed by those users with network issues.