diff --git a/paddlepalm/head/match.py b/paddlepalm/head/match.py index 7742f4a11aff065450fc926f487aa9d48537c57f..9df4a1a1d8c532db94eaf3484ac88581184e07b8 100644 --- a/paddlepalm/head/match.py +++ b/paddlepalm/head/match.py @@ -182,7 +182,7 @@ class Match(Head): label = np.argmax(np.array(self._preds[i])) result = {'index': i, 'label': label, 'logits': self._preds_logits[i], 'probs': self._preds[i]} elif self._learning_strategy == 'pairwise': - result = {'index': i, 'label': label, 'probs': self._preds[i][0]} + result = {'index': i, 'probs': self._preds[i][0]} result = json.dumps(result, ensure_ascii=False) writer.write(result+'\n') print('Predictions saved at '+os.path.join(output_dir, 'predictions.json'))