错误:nvalid 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
分析:版本更新问题. pytorch0.3.1版本代码,在0.4-0.5版本的pytorch会出现警告,不会报错,但是0.5版本以上的pytorch就会报错
解决方法:
将原语句:class_cost_vector.append(class_loss.data[0])
修改为:class_cost_vector.append(class_loss.item())