소스 검색

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()