提交 978fd680 编写于 作者: Q Qiao Longfei

update ctr_reader.py

上级 7f07dfa1
...@@ -90,7 +90,7 @@ def ctr_reader(feed_data, ...@@ -90,7 +90,7 @@ def ctr_reader(feed_data,
reader_name = "_".join([name, "reader"]) reader_name = "_".join([name, "reader"])
var = global_scope().var(queue_name) var = global_scope().var(queue_name)
feed_queue = core.init_lod_tensor_blocking_queue(var, capacity, shapes) feed_queue = core.init_lod_tensor_blocking_queue(var, capacity)
startup_blk = default_startup_program().current_block() startup_blk = default_startup_program().current_block()
reader_var = startup_blk.create_var(name=reader_name) reader_var = startup_blk.create_var(name=reader_name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册