-
카테고리
-
세부 분야
딥러닝 · 머신러닝
-
해결 여부
해결됨
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)
답변을 작성해보세요.
0
답변 1