label vec to text vec (#3541)

facebookresearch · Mar 19, 2021 · 455a567 · 455a567
1 parent 1a3ffe0
commit 455a567
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/parlai/core/torch_generator_agent.py b/parlai/core/torch_generator_agent.py
@@ -844,7 +844,7 @@ def rank_eval_label_candidates(self, batch, batchsize):
             num_cands = len(batch.candidate_vecs[i])
             enc = self.model.reorder_encoder_states(encoder_states, [i] * num_cands)
             cands, _ = self._pad_tensor(batch.candidate_vecs[i])
-            cands = cands.to(batch.label_vec.device)
+            cands = cands.to(batch.text_vec.device)
             scores, _ = self.model.decode_forced(enc, cands)
             score_view = scores.reshape(num_cands * cands.size(1), -1)
             cand_losses = F.cross_entropy(