Inflearn Community Q&A
asked
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
