fix cls type

66c3294c · littletomatodonkey · a81b88a0 · 66c3294c
Commit 66c3294c authored Aug 17, 2021 by littletomatodonkey
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/ppocr/losses/cls_loss.py
+++ b/ppocr/losses/cls_loss.py
@@ -25,6 +25,6 @@ class ClsLoss(nn.Layer):
        self.loss_func = nn.CrossEntropyLoss(reduction='mean')

    def forward(self, predicts, batch):
-        label = batch[1]
+        label = batch[1].astype("int64")
        loss = self.loss_func(input=predicts, label=label)
        return {'loss': loss}