딥러닝 모델의 학습 과정은 일반적으로 모델 정의
-> Optimizer 정의
-> Loss 함수 정의
-> 학습을 위한 데이터셋 정의
-> 학습을 위한 데이터로더 생성
-> 모델 학습
순으로 이루어진다. 본 강에서는 모델 정의
-> Optimizer 정의
-> Loss 함수 정의
부분에 해당하는 AutoGrad & Optimizer와 학습을 위한 데이터셋 정의
-> 학습을 위한 데이터로더 생성
부분에 해당하는 DataSets & DataLoaders 에 대하여 학습을 진행했다.
class Layer(nn.Module):
def __init__(self):
super(Layer, self).__init()
'''
파라미터 초기화 부분
'''
def forward(self):
'''
순전파 계산
'''
def backward(self):
'''
역전파 계산
'''
def optimize(self):
'''
파라미터 업데이트
'''
class CustomDataset(Dataset):
def __init__(self):
'''
변수 설정 및 데이터 전처리
'''
def __len__(self):
'''
전체 데이터의 개수 반환
'''
def __getitem__(self, idx):
'''
input idx에 해당하는 데이터 반환
'''