From 182688d510fcc1182ea4d11bb18f2ccfad8f3436 Mon Sep 17 00:00:00 2001 From: wangxiao Date: Wed, 4 Dec 2019 17:01:55 +0800 Subject: [PATCH] fix unique_ids --- paddlepalm/task_paradigm/mrc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddlepalm/task_paradigm/mrc.py b/paddlepalm/task_paradigm/mrc.py index 2a114c5..c4d8662 100644 --- a/paddlepalm/task_paradigm/mrc.py +++ b/paddlepalm/task_paradigm/mrc.py @@ -121,7 +121,7 @@ class TaskParadigm(task_paradigm): def postprocess(self, rt_outputs): """this func will be called after each step(batch) of training/evaluating/predicting process.""" if not self._is_training: - unique_ids = np.squeeze(rt_outputs['unique_ids'], -1) + unique_ids = rt_outputs['unique_ids'] start_logits = rt_outputs['start_logits'] end_logits = rt_outputs['end_logits'] for idx in range(len(unique_ids)): -- GitLab