瀏覽代碼

Suppress scalar teensor warning

Bobholamovic 1 年之前
父節點
當前提交
600caff5b1
共有 1 個文件被更改,包括 6 次插入1 次删除
  1. 6 1
      paddlers/utils/stats.py

+ 6 - 1
paddlers/utils/stats.py

@@ -49,7 +49,12 @@ class TrainingStats(object):
                 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()