AI
-
[pytorch 튜토리얼] 2. 모델 구현 및 저장AI/Pytorch 2021. 10. 13. 18:51
1. 모델 매개변수 최적화하기 기존에 작성한 모델을 사용해 Fashion Data를 학습해볼 것임. 매개변수를 최적화하여 모델을 학습하고 테스트함. 학습 과정은 반복적이며 매 횟수를 epoch이라 부르고 추측과 정답의 loss값을 계산해 역전파하여 학습함(이 때 파라미터가 최적화optimize됨) 1.1 기본(pre-requisite) 코드 데이터셋 클래스 생성 import torch from torch import nn from torch.utils.data import DataLoader from torchvision import datasets from torchvision.transforms import ToTensor, Lambda training_data = datasets.FashionMNI..
-
[Pytorch 튜토리얼] 1. 파이토치 기본 익히기AI/Pytorch 2021. 10. 13. 07:48
0. 텐서 텐서는 배열이나 행렬과 매우 유사함. PyTorch에서는 연산을 할 때 텐서를 이용함. 텐서는 Numpy의 ndarray와 유사하지만 GPU나 하드웨어 가속기에서 실행 가능하다는 점이 특징. (상호 변환도 가능) 1. Dataset과 DataLoader 데이터 처리 코드가 너무 복잡하기에 가독성과 모듈화를 위해 데이터셋 코드는 따로 빼두어야 함. Pytorch에서는 자체적으로 torch.utils.data.DataLoader와 torch.utils.data.Dataset 두 가지의 api를 제공해 미리 준비된 데이터셋 뿐만 아니라 내가 가지고 있는 자체 데이터를 사용하기 쉽게 해준다. Dataset은 샘플과 label을 저장하고, DataLoader는 Dataset을 샘플에 쉽게 접근할 수 있..