Skip to content

GitLab

Unverified Commit 13757fe1 authored by xiaoting's avatar xiaoting Committed by GitHub
Browse files

Merge pull request #4495 from andyjpaddle/fix_sar_cpu_train 

fix sar train on cpu
parents 28b52539 92c9eaf3
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
ppocr/modeling/heads/rec_sar_head.py
View file @ 13757fe1
...@@ -275,7 +275,6 @@ class ParallelSARDecoder(BaseDecoder): ...@@ -275,7 +275,6 @@ class ParallelSARDecoder(BaseDecoder):
if img_metas is not None and self.mask: if img_metas is not None and self.mask:
valid_ratios = img_metas[-1] valid_ratios = img_metas[-1]
label = label.cuda()
lab_embedding = self.embedding(label) lab_embedding = self.embedding(label)
# bsz * seq_len * emb_dim # bsz * seq_len * emb_dim
out_enc = out_enc.unsqueeze(1) out_enc = out_enc.unsqueeze(1)
......
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