IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item<T>()` in C++ to convert a 0-dim tensor to a number
pytorch在训练时发生如上报错,有其他大佬解释是pytorch版本问题造成的.data[0]和.item()的问题,但我在创建虚拟环境,安装其它版本pytorch后依旧出现同样的错误,因此排除了pytorch版本问题。
出错位置如下:
如下:
文章出处登录后可见!
已经登录?立即刷新