L1aoXingyu/code-of-learn-deep-learning-with-pytorch

第五章 rnn用于时间序列的分析

WenbinSun opened this issue · 2 comments

WenbinSun commented

请问老师这个问题怎么解决觉得？

if (e + 1) % 100 == 0: # 每 100 次输出结果
---> 12 print('Epoch: {}, Loss: {:.5f}'.format(e + 1, loss.data[0]))

IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

L1aoXingyu commented

把 loss.data[0] 改成 loss.item() 就可以了，这是版本不一致的问题

  • 😄1
  • ❤️1
WenbinSun commented

把 loss.data[0] 改成 loss.item() 就可以了，这是版本不一致的问题

谢谢老师。