[debugging] torchvison model 다운로드시 나타나는 WinError 10060 해결하기

torchvision에서 transfer learning model을 불러와 학습시키고자 할 때 해당 에러가 나올 수 있다.

model = models.resnet50(pretrained=True).to(device)

이렇게 모델을 가져오면, https://download.pytorch.org/models/모델명 링크를 통해 가져온다.

이 때 WinError 10060 이 발생한다면, 직접 다운로드 받아 경로에 옮겨 모델을 사용할 수 있다.

torch.hub.get_dir()

1. 먼저, 위 코드를 입력해 어떤 경로에 다운로드 받아야하는지 확인한다.

보통 로컬의 경우 'C://Users//사용자이름//.cache//torch//hub//checkpoints' 일 것이다. (윈도우)

2. 확인한 후에 https://download.pytorch.org/models/모델명 해당 링크를 브라우저에 입력해 직접 다운로드 받는다.

3. 받은 .pth 파일(모델 파일)을 해당 위치에 옮겨서 에러가 났던 코드를 다시 실행하면 무리없이 사용할 수 있다!