Suppress scalar teensor warning

PaddlePaddle · Aug 17, 2023 · 600caff · 600caff
1 parent 42ac0a9
commit 600caff
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/paddlers/utils/stats.py b/paddlers/utils/stats.py
@@ -49,7 +49,12 @@ def update(self, stats):
                 for k in stats.keys()
             }
         for k, v in self.meters.items():
-            v.update(stats[k].numpy())
+            stat = stats[k]
+            if stat.ndim == 0:
+                stat = float(stat)
+            else:
+                stat = stat.numpy()
+            v.update(stat)
 
     def get(self, extras=None):
         stats = collections.OrderedDict()