optimization1 [파이토치]최적화 이번 포스트에선 모델의 파라미터들을 최적화하는법에 대해 알아보고자 한다. 모델을 학습하는 과정은 주어진 epoch 수만큼 반복을 하며 출력값을 추정하고 loss 값을 계산한다. 그리고 경사하강법을 통해 모델이 최적의 성능을 내도록 파라미터를 최적화한다. 1. 신경망 구성하기 파라미터를 최적화할 기초적인 수준의 신경망을 만든다. 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.FashionMNIST( root="data", tra.. 2022. 10. 6. 이전 1 다음
