Don't do softmax

yzhangcs · yzhangcs · commit e890e41e397e · 2021-04-27T15:57:54.000+08:00
diff --git a/supar/parsers/sdp.py b/supar/parsers/sdp.py
@@ -454,7 +454,7 @@ def _predict(self, loader):
             label_preds = self.model.decode(s_edge, s_label).masked_fill(~mask, -1)
             preds['labels'].extend(chart[1:i, :i].tolist() for i, chart in zip(lens, label_preds))
             if self.args.prob:
-                preds['probs'].extend([prob[1:i, :i].cpu() for i, prob in zip(lens, s_edge.softmax(-1).unbind())])
+                preds['probs'].extend([prob[1:i, :i].cpu() for i, prob in zip(lens, s_edge.unbind())])
         preds['labels'] = [CoNLL.build_relations([[self.LABEL.vocab[i] if i >= 0 else None for i in row] for row in chart])
                            for chart in preds['labels']]
 
