해결된 질문
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되었군요. 확인해주시고 해결책까지 공유해주셔서 감사합니다!
강의 자료에도 반영하겠습니다!