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

update for new version of torch

Browse Source
This commit is contained in:
morvanzhou
2018-11-13 10:26:34 +08:00
parent 0f4219c72c
commit 3e2d0b8a40
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tutorial-contents-notebooks/502_GPU.ipynb
View File

@ -213,7 +213,7 @@
" # !!!!!!!! Change in here !!!!!!!!! #\n", " # !!!!!!!! Change in here !!!!!!!!! #\n",
" pred_y = torch.max(test_output, 1)[1].cuda().data.squeeze() # move the computation in GPU\n", " pred_y = torch.max(test_output, 1)[1].cuda().data.squeeze() # move the computation in GPU\n",
"\n", "\n",
" accuracy = sum(pred_y == test_y) / test_y.size(0)\n", " accuracy = torch.sum(pred_y == test_y).type(torch.FloatTensor) / test_y.size(0)\n",
" print('Epoch: ', epoch, '| train loss: %.4f' % loss.data[0], '| test accuracy: %.2f' % accuracy)\n" " print('Epoch: ', epoch, '| train loss: %.4f' % loss.data[0], '| test accuracy: %.2f' % accuracy)\n"
] ]
}, },