강의

멘토링

로드맵

Inflearn Community Q&A

mysticize911785's profile image
mysticize911785

asked

Complete mastery of deep learning theory + PyTorch practice

[Lab] Model Save/Load

from torchvision.models.resnet import model_urls 에러

Resolved

Written on

·

848

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)
딥러닝pytorchNLP컴퓨터-비전cnnrnntransformerllm

Quiz

PyTorch에서 DataLoader의 주된 역할은 무엇일까요?

모델 구조를 정의합니다.

데이터셋을 미니배치로 묶어 제공합니다.

손실 함수 값을 계산합니다.

모델 가중치를 초기화합니다.

Answer 1

0

peterbyun969574님의 프로필 이미지
peterbyun969574
Instructor

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

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

mysticize911785's profile image
mysticize911785

asked

Ask a question