Skip to content

Commit 6e27f8f

Browse files
authored
fix log print with loss inf (#645)
1 parent 71046d5 commit 6e27f8f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

mindocr/utils/misc.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,8 @@ def reset(self) -> None:
2222
self.count = Tensor(0.0, dtype=ms.float32)
2323

2424
def update(self, val: Tensor, n: int = 1) -> None:
25+
if val == float("inf"):
26+
return
2527
self.val = val
2628
self.sum += val * n
2729
self.count += n

0 commit comments

Comments
 (0)