Merge pull request #4495 from andyjpaddle/fix_sar_cpu_train

fix sar train on cpu

Merge pull request #4495 from andyjpaddle/fix_sar_cpu_train
fix sar train on cpu
13757fe1 · xiaoting · GitHub · 28b52539 · 92c9eaf3 · 13757fe1
Unverified Commit 13757fe1 authored Oct 29, 2021 by xiaoting Committed by GitHub Oct 29, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

ppocr/modeling/heads/rec_sar_head.py ppocr/modeling/heads/rec_sar_head.py +0 -1

No files found.
--- a/ppocr/modeling/heads/rec_sar_head.py
+++ b/ppocr/modeling/heads/rec_sar_head.py
@@ -275,7 +275,6 @@ class ParallelSARDecoder(BaseDecoder):
        if img_metas is not None and self.mask:
            valid_ratios = img_metas[-1]

-        label = label.cuda()
        lab_embedding = self.embedding(label)
        # bsz * seq_len * emb_dim
        out_enc = out_enc.unsqueeze(1)