- 博客(4)
- 收藏
- 关注
原创 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() 方法是用
2022-05-17 20:58:04
1312
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人