fix <not serializable> (#4654)

facebookresearch · Jul 13, 2022 · f706af2 · f706af2
1 parent 03ca319
commit f706af2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/parlai/core/torch_classifier_agent.py b/parlai/core/torch_classifier_agent.py
@@ -703,7 +703,7 @@ def eval_step(self, batch):
             self._update_confusion_matrix(batch, preds)
 
         if self.opt.get('print_scores', False):
-            return Output(preds, class_list=[self.class_list], probs=probs.cpu())
+            return Output(preds, class_list=[self.class_list], probs=probs.tolist())
         if self.opt.get('return_cand_scores', False):
             sorted_scores, ranks = probs.sort(1, descending=True)
             sorted_scores = sorted_scores.cpu()