머신러닝 및 딥러닝/Pytorch GO_IN 2021. 5. 28. 15:35
지난번에 SGD 옵티마이저를 사용하여 MNIST 데이터 셋을 학습시키는 과정에 대해 살펴보았습니다. 이번에는 SGD를 포함해서 Adagrad, RMSprop, Adam 옵티마이저를 사용하여 MNIST 데이터 셋 학습을 진행해보려고 합니다. 코드 중간중간에 주석이 있기는 하지만 지난번 포스팅과 차이가 있는 부분만 설명드리려고 하니 조금 더 자세한 내용이 궁금하시다면 화면 하단의 SGD로 MNIST 데이터 셋 학습 링크를 참고해주시기 바랍니다. 또한 화면 하단 github 주소에 전체 코드를 올려두겠습니다. 1. 모듈 및 라이브러리 import import torch.nn as nn import torch import matplotlib.pyplot as plt import numpy as np from t..
더 읽기
머신러닝 및 딥러닝/Pytorch GO_IN 2021. 4. 26. 16:03
이번 글에서는 Pytorch를 사용하여 jupyter notebook에서 MNIST 데이터 셋을 학습하는 것에 대해 알아보려고 합니다. 모델을 구성하여 학습을 시키고, 최종적으로 epoch에 따른 loss와 정확도를 matplotlib을 이용해서 그래프를 그려보려고 합니다. 전체 코드는 제일 하단 부분에 github 사이트로 들어가시면 됩니다. 1. 관련 모듈 및 라이브러리 import import torch.nn as nn import torch from torchvision import datasets from torchvision import transforms from torch.utils.data import DataLoader import numpy as np import torch.optim..
