From 0955e2d867a445f5d646209e952e79bd2d3cb83d Mon Sep 17 00:00:00 2001
From: wangxiao <zko1021@foxmail.com>
Date: Wed, 4 Dec 2019 21:57:06 +0800
Subject: [PATCH] fix bugs

---
 paddlepalm/utils/reader_helper.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddlepalm/utils/reader_helper.py b/paddlepalm/utils/reader_helper.py
index 09d5c0c..d915179 100644
--- a/paddlepalm/utils/reader_helper.py
+++ b/paddlepalm/utils/reader_helper.py
@@ -218,7 +218,7 @@ def merge_input_attrs(backbone_attr, task_attrs, insert_taskid=True, insert_batc
     names = []
     start = 0
     if insert_taskid:
-        ret.append(([1], 'int64'))
+        ret.append(([1, 1], 'int64'))
         names.append('__task_id')
         start += 1
     
-- 
GitLab