인프런 커뮤니티 질문&답변

첨부된 9-5 ipynb 에 resnet weight 를 불러올 때

from torchvision.models.resnet import model_urls

에서 오류가 나와 찾아보니 torchvision 0.13 버전 이후에

model_urls 는 없어졌다고 하네요.

https://github.com/pytorch/vision/blob/main/torchvision/models/resnet.py

아래와 같이 해결을 해보았습니다.

from torchvision.models.resnet import ResNet50_Weights

checkpoint = load_state_dict_from_url(ResNet50_Weights.IMAGENET1K_V2.url)