Merge pull request #4463 from Topdu/dygraph

modify data type to adaptive

Merge pull request #4463 from Topdu/dygraph
modify data type to adaptive
f96797e1 · xiaoting · GitHub · bbf4625e · 81f0a83e · f96797e1
Unverified Commit f96797e1 authored Oct 28, 2021 by xiaoting Committed by GitHub Oct 28, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

ppocr/losses/rec_nrtr_loss.py ppocr/losses/rec_nrtr_loss.py +1 -1

No files found.
--- a/ppocr/losses/rec_nrtr_loss.py
+++ b/ppocr/losses/rec_nrtr_loss.py
@@ -22,7 +22,7 @@ class NRTRLoss(nn.Layer):
            log_prb = F.log_softmax(pred, axis=1)
            non_pad_mask = paddle.not_equal(
                tgt, paddle.zeros(
-                    tgt.shape, dtype='int64'))
+                    tgt.shape, dtype=tgt.dtype))
            loss = -(one_hot * log_prb).sum(axis=1)
            loss = loss.masked_select(non_pad_mask).mean()
        else: