python的item()用法

在看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张量转化为标量的方法。

转载自

python 的 iter()、item()、items()三种方法_一直努力的喜羊羊的博客-CSDN博客

你可能感兴趣的:(python,深度学习,计算机视觉)