Skip to content

GitLab

Unverified Commit 49382181 authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Merge pull request #3717 from littletomatodonkey/dyg/fix_cls_type 

fix cls type
parents ab16f2e4 66c3294c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ppocr/losses/cls_loss.py
View file @ 49382181
...@@ -25,6 +25,6 @@ class ClsLoss(nn.Layer): ...@@ -25,6 +25,6 @@ class ClsLoss(nn.Layer):
self.loss_func = nn.CrossEntropyLoss(reduction='mean') self.loss_func = nn.CrossEntropyLoss(reduction='mean')
def forward(self, predicts, batch): def forward(self, predicts, batch):
label = batch[1] label = batch[1].astype("int64")
loss = self.loss_func(input=predicts, label=label) loss = self.loss_func(input=predicts, label=label)
return {'loss': loss} return {'loss': loss}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment