딥러닝 모델의 학습 과정은 일반적으로 모델 정의 -> 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에 해당하는 데이터 반환
'''