다음은 모델의 파라미터를 정의하는 예시 코드입니다.
from torch.optim import optimizer criterion = torch.nn.MSELoss() optimizer = torch.optim.SGD(model.parameters(), lr=0.01, momentum=0.9) scheduler = torch.optim.lr_scheduler.LambdaLR(optimizer=optimizer, lr_lambda=lambda epoch: 0.95 ** epoch) for epoch in range(1, 100+1): ------ 에포크 수만큼 데이터를 반복하여 처리 for x, y in dataloader: ------ 배치 크기만큼 데이터를 가져와서 학습 진행 optimizer.zero_grad() loss_fn(model(x), y).backward() optimizer.step() scheduler.step()