IndexError:too many indices for array

在用pytorch进行深度学习训练时，出现如下错误：

经过查询，发现该错误是由于pytorch升级引起的，我使用的版本为pytorch 0.40版本，所以做了如下修改，问题解决

if torch.__version__.startswith('0.3'): 

    bbox_loss += net.bbox_loss.data.cpu().numpy()[0]

    iou_loss += net.iou_loss.data.cpu().numpy()[0]

    cls_loss += net.cls_loss.data.cpu().numpy()[0]

    train_loss += loss.data.cpu().numpy()[0]

else:

    bbox_loss += float(net.bbox_loss.data.cpu().numpy())

    iou_loss += float(net.iou_loss.data.cpu().numpy())

    cls_loss += float(net.cls_loss.data.cpu().numpy())

    train_loss += float(loss.data.cpu().numpy())

或者

在pytorch 0.4.0中使用net.bbox_loss.item()