未验证 提交 03f8464d 编写于 作者: L Lin Manhui 提交者: GitHub

fix ppdet/utils/states (#8421) (#8496)

* fix a code

* fix the tensor to float
Co-authored-by: Nluyao-cv <1367355728@qq.com>
上级 47d7ee5a
......@@ -74,7 +74,7 @@ class TrainingStats(object):
for k in stats.keys()
}
for k, v in self.meters.items():
v.update(stats[k].numpy())
v.update(float(stats[k]))
def get(self, extras=None):
stats = collections.OrderedDict()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册