提交 7e967e03 编写于 作者: W wuzewu

Fix parallel issue

上级 a585092b
......@@ -176,6 +176,10 @@ class DetectionTask(BaseTask):
shape=feed_var_map[key]['shape'],
dtype=feed_var_map[key]['dtype'],
lod_level=feed_var_map[key]['lod_level'])
# Add assign operation to prevent the im_id from being optimized and cause the exception
if feed_var_map[key]['name'] == 'im_id':
_ = fluid.layers.assign(l)
labels.append(l)
return labels
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册