• 카테고리

    질문 & 답변
  • 세부 분야

    딥러닝 · 머신러닝

  • 해결 여부

    해결됨

from torchvision.models.resnet import model_urls 에러

24.01.29 22:23 작성 조회수 244

2

첨부된 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)

답변 1

답변을 작성해보세요.

0

앗 model_urls가 deprecation되었군요. 확인해주시고 해결책까지 공유해주셔서 감사합니다!

강의 자료에도 반영하겠습니다!