未验证 提交 fccdf3c9 编写于 作者: Y Yibing Liu 提交者: GitHub

Merge pull request #808 from kuke/fix_test

Fix data feed in test & profile
...@@ -168,7 +168,7 @@ def profile(args): ...@@ -168,7 +168,7 @@ def profile(args):
start_time = time.time() start_time = time.time()
frames_seen = 0 frames_seen = 0
# load_data # load_data
(features, labels, lod) = batch_data (features, labels, lod, _) = batch_data
feature_t.set(features, place) feature_t.set(features, place)
feature_t.set_lod([lod]) feature_t.set_lod([lod])
label_t.set(labels, place) label_t.set(labels, place)
......
...@@ -192,7 +192,7 @@ def train(args): ...@@ -192,7 +192,7 @@ def train(args):
test_data_reader.batch_iterator(args.batch_size, test_data_reader.batch_iterator(args.batch_size,
args.minimum_batch_size)): args.minimum_batch_size)):
# load_data # load_data
(features, labels, lod) = batch_data (features, labels, lod, _) = batch_data
feature_t.set(features, place) feature_t.set(features, place)
feature_t.set_lod([lod]) feature_t.set_lod([lod])
label_t.set(labels, place) label_t.set(labels, place)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册