3장 p96 오타 - Githubissues

keon / 3-min-pytorch

<펭귄브로의 3분 딥러닝, 파이토치맛> 예제 코드

MIT License

208 stars 246 forks source link

Open ysangj opened 5 years ago

ysangj commented 5 years ago

96 페이지의 코드블락을 수정해야 합니다.

현재:

model.eval()
test_loss_before = criterion(torch.squeeze(model(x_test)), y_test)
print('After Training, test loss is {}'.format(test_loss.item()))

수정:

model.eval()
test_loss = criterion(torch.squeeze(model(x_test)), y_test)
print('After Training, test loss is {}'.format(test_loss.item()))