Forráskód Böngészése

Suppress scalar teensor warning

Bobholamovic 1 éve
szülő
commit
600caff5b1
1 módosított fájl, 6 hozzáadás és 1 törlés
  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()