Machine Learning/PyTorch
-
[PyTorch/에러] RuntimeError: Error(s) in loading state_dictMachine Learning/PyTorch 2022. 12. 18. 00:55
pretrain 된 모델을 불러와 추가 데이터에 대해 finetuning 할 때 더 좋은 성능을 보여주는 경우가 많습니다. 이때 load_state_dict 을 이용해 모델을 load 하는데, 사전에 정의한 모델과 불러오려는 모델의 state_dict - key 가 다를 경우 아래와 같은 에러가 발생할 수 있습니다. RuntimeError: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict: "conv1.weight", "bn1.weight", "bn1.bias", "bn1.running_mean", "bn1.running_var", "layer1.0.conv1.weight", "layer1.0.bn1.weight", "lay..
-
[PyTorch] Deep learning with PyTorch - IntroMachine Learning/PyTorch 2022. 4. 27. 13:12
PyTorch로 딥러닝 하기: 60분 만에 끝장내기의 초간단 정리입니다. 학습 및 정리 과정에서 들었던 의문이나 궁금한 사항에 대한 답도 설명에 추가했습니다. 실습은 구글 코랩(Google Colab) 을 통해 쉽게 따라 하실 수 있습니다. Pytorch Python 기반의 연산 패키지, 오픈 소스 Machine Learning 프레임워크 GPU 이용 연산 필요한 경우 사용 (Numpy 대체) 신경망 구현에 유용한 라이브러리 제공 TensorFlow와 양대산맥을 이루는 딥러닝 연구 플랫폼 / 실제 서비스와 프로덕션 배포에도 사용 Tensors 배열이나 행렬과 유사한 자료구조 / Numpy의 ndarray와 유사 / GPU를 사용한 연산 가속 가능 import torch import numpy as np ..