Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

fix <not serializable> in torch_classifier agent world log #4654

Merged
merged 1 commit into from Jul 13, 2022
Merged

fix <not serializable> in torch_classifier agent world log #4654

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
2 changes: 1 addition & 1 deletion parlai/core/torch_classifier_agent.py
View file
Open in desktop
Expand Up @@ -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()
Expand Down