在看faster-rcnn源码时候发现一段代码为
loss_dict_reduced = utils.reduce_dict(loss_dict)
losses_reduced = sum(loss for loss in loss_dict_reduced.values())
loss_value = losses_reduced.item()
是每一次训练时对loss的计算,原以为loss是个字典,里面的item()用法是items()用法,后来发现并不是。
其实是,item() 方法是用来将只有一个元素的numpy数组或tensor张量转化为标量的方法。
转载自