fix ppdet/utils/states (PaddlePaddle#8421)

* fix a code * fix the tensor to float
Bobholamovic · Jul 31, 2023 · 0f739d8 · 0f739d8
1 parent 47d7ee5
commit 0f739d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppdet/utils/stats.py b/ppdet/utils/stats.py
@@ -74,7 +74,7 @@ def update(self, stats):
                 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()