fix bugs

0955e2d8 · wangxiao · 935909d8 · 0955e2d8
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddlepalm/utils/reader_helper.py paddlepalm/utils/reader_helper.py +1 -1

未找到文件。
--- 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