modify int64 to int32 for windows train

2021-10-27 15:53:46 +08:00 · 2021-10-27 15:53:46 +08:00 · b995249965
parent 1008376c3f
commit b995249965
1 changed files with 1 additions and 1 deletions
--- 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='int32'))
            loss = -(one_hot * log_prb).sum(axis=1)
            loss = loss.masked_select(non_pad_mask).mean()
        else: