fix cls type

bupt906 · Aug 17, 2021 · 66c3294 · 66c3294
1 parent a81b88a
commit 66c3294
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppocr/losses/cls_loss.py b/ppocr/losses/cls_loss.py
@@ -25,6 +25,6 @@ def __init__(self, **kwargs):
         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}