Open ysangj opened 4 years ago
test_loss_before = criterion(torch.squeeze(model(x_test)), y_test)
를 다음과 같이 바꿔 줘야 합니다.
test_loss = criterion(model(x_test).squeeze(), y_test)
베타 리더들의 피드백 중 하나였던 이슈입니다. 전체 코드에는 이미 수정이 이미 되있습니다.
test_loss_before = criterion(torch.squeeze(model(x_test)), y_test)를 다음과 같이 바꿔 줘야 합니다.
test_loss = criterion(model(x_test).squeeze(), y_test)베타 리더들의 피드백 중 하나였던 이슈입니다. 전체 코드에는 이미 수정이 이미 되있습니다.