[Pytorch] model.eval()
·
언어 | Framework/Pytorch
Pytorch를 사용해서 학습을 할 때, 보통 다음과 같이 구성한다. #......... for epoch in range(0, epoch_num): for data, label in data_loader: model.train()### train setting output = model(data) ### # 학습 과정... ### model.eval()### evaluation setting with torch.no_grad(): ### # 평가 과정... ### #......... 위 코드에서 model.train()은 train 할 때, model.eval()은 evaluation(평가, val_loss 계산) 할 때 앞서 실행한다고 알고 있다. 정확히 무슨 동작을 하는 것인가... Pytorch ..
woongs_93
'model.train()' 태그의 글 목록