update for new version of torch

This commit is contained in:
morvanzhou
2018-11-08 19:40:27 +08:00
parent 906cf71b6f
commit bf4cf1e700
2 changed files with 11 additions and 3 deletions

View File

@ -249,7 +249,7 @@
" opt.zero_grad() # clear gradients for next train\n",
" loss.backward() # backpropagation, compute gradients\n",
" opt.step() # apply gradients\n",
" l_his.append(loss.data[0]) # loss recoder\n",
" l_his.append(loss.item()) # loss recoder\n",
"\n",
"labels = ['SGD', 'Momentum', 'RMSprop', 'Adam']\n",
"for i, l_his in enumerate(losses_his):\n",