Skip to content

Commit

Permalink
fix the tensor to float
Browse files Browse the repository at this point in the history
  • Loading branch information
luyao-cv committed Jul 10, 2023
1 parent 50d7972 commit 420dd76
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ppdet/utils/stats.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down

0 comments on commit 420dd76

Please sign in to comment.