IndexError: The shape of the mask [32, 8732] at index 0does not match the shape of the indexed tensor [279424, 1] at index 0
如果出现类似于以上的索引错误,需要将layers/modules/multibox_loss.py 的代码按照下图修改即可。注释掉的是源代码,下面的是新加的,即交换两行代码的顺序即可。
解决了这个问题,又会出现一个新问题:
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
解决方法:将错误行的 .data[0] 修改为 .item() 即可
代码语言:javascript复制 # loc_loss = loss_l.data[0]
# conf_loss = loss_c.data[0]
loc_loss = loss_l.item()
conf_loss = loss_c.item()