Merge pull request #4495 from andyjpaddle/fix_sar_cpu_train

fix sar train on cpu
2026-09-24 23:33:08 +08:00 · 2021-10-29 22:18:11 +08:00
parent 28b525397d 92c9eaf314
commit 13757fe1cb
1 changed files with 0 additions and 1 deletions
@@ -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)