fix the tensor to float

luyao-cv · luyao-cv · commit 420dd76192a8 · 2023-07-10T08:38:35.000Z
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])
+            v.update(float(stats[k]))
 
     def get(self, extras=None):
         stats = collections.OrderedDict()

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,7 @@ def update(self, stats):`
`74`	`74`	`for k in stats.keys()`
`75`	`75`	`}`
`76`	`76`	`for k, v in self.meters.items():`
`77`		`- v.update(stats[k])`
	`77`	`+ v.update(float(stats[k]))`
`78`	`78`
`79`	`79`	`def get(self, extras=None):`
`80`	`80`	`stats = collections.OrderedDict()`