신경망 모델1 [개념 정리] 신경망 모델 정의하기 Pytorch로 신경망 모델을 정의할 때 생각해야 하는 것 3가지 Model 디자인 loss와 최적화 설계 Train Cycle (forward, backward, update) 첫 번째로 Class와 변수 정의 방법을 알아보자 사용자 정의 nn 모듈 nn.Module을 상속한 Class 이용 Pytorch 모델의 기본 구조 import torch import torch.nn as nn import torch.nn.functional as F class Model_Name(nn.Module): def __init__(self): super(Model_Name, self).__init__() self.conv1 = nn.Conv2d() self.conv2 = nn.Conv2d() def forward(se.. 2022. 3. 31. 이전 1 다음