解决IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python 的问题
问题原因 pytorch版本问题
解决:.data[0]改成.item
解决IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python 的问题
问题原因 pytorch版本问题
解决:.data[0]改成.item