diff --git a/ppdet/utils/stats.py b/ppdet/utils/stats.py index 4cd36d91cf80418720c24915522f1cf4587fe7bd..c070e6544ed9e25b2ed156156ec8a6379717ebc9 100644 --- a/ppdet/utils/stats.py +++ b/ppdet/utils/stats.py @@ -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()