MNIST1 MNIST 데이터 학습 모델 구현 이번 포스팅에서는 MNIST손글씨 데이터셋 학습모델을 Pytorch로 구현해보겠습니다. 총 에포크 수를 5번으로 제한하고 정확도를 최대한 높이는 식으로 진행해 보았습니다. - 소스코드 import torch import torch.nn as nn import torch.optim as optim import torchvision.datasets as dset import torchvision.transforms as transforms import matplotlib.pyplot as plt from torch.utils.data import DataLoader from torch.optim import lr_scheduler from torchsummary import summary train_tran.. 2023. 7. 13. 이전 1 다음
