pytorch版本更新问题

错误：nvalid index of a 0-dim tensor. Use tensor.item() in Python or tensor.item() 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())