andre/PyTorch-Tutorial
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #5 from liufuyang/fix-up-401

Browse Source 
Fix up test code
This commit is contained in:
Morvan
2017-06-21 18:12:34 +10:00
committed by GitHub
parent 51f1c938f3 528d7ccd41
commit 1adcbb999f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tutorial-contents/401_CNN.py
View File

@ -122,7 +122,7 @@ for epoch in range(EPOCH):
plt.ioff() plt.ioff()
# print 10 predictions from test data # print 10 predictions from test data
test_output = cnn(test_x[:10]) test_output, _ = cnn(test_x[:10])
pred_y = torch.max(test_output, 1)[1].data.numpy().squeeze() pred_y = torch.max(test_output, 1)[1].data.numpy().squeeze()
print(pred_y, 'prediction number') print(pred_y, 'prediction number')
print(test_y[:10].numpy(), 'real number') print(test_y[:10].numpy(), 'real number')