인프런 커뮤니티 질문&답변
작성한 질문수
from torchvision.models.resnet import model_urls 에러
해결된 질문
24.01.29 22:23 작성
·
584
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
변정현
지식공유자
2024. 01. 29. 23:35
앗 model_urls가 deprecation되었군요. 확인해주시고 해결책까지 공유해주셔서 감사합니다!
강의 자료에도 반영하겠습니다!