提交 609f7aa6 编写于 作者: D dangqingqing

Fix the test reader.

上级 32481e12
...@@ -174,7 +174,7 @@ def train(train_file_list, ...@@ -174,7 +174,7 @@ def train(train_file_list,
train_reader = paddle.batch( train_reader = paddle.batch(
reader.train(data_args, train_file_list), batch_size=batch_size) reader.train(data_args, train_file_list), batch_size=batch_size)
test_reader = paddle.batch( test_reader = paddle.batch(
reader.test(data_args, train_file_list), batch_size=batch_size) reader.test(data_args, val_file_list), batch_size=batch_size)
feeder = fluid.DataFeeder( feeder = fluid.DataFeeder(
place=place, feed_list=[image, gt_box, gt_label, difficult]) place=place, feed_list=[image, gt_box, gt_label, difficult])
...@@ -217,6 +217,6 @@ if __name__ == '__main__': ...@@ -217,6 +217,6 @@ if __name__ == '__main__':
train_file_list='./data/trainval.txt', train_file_list='./data/trainval.txt',
val_file_list='./data/test.txt', val_file_list='./data/test.txt',
data_args=data_args, data_args=data_args,
learning_rate=0.004, learning_rate=0.001,
batch_size=32, batch_size=32,
num_passes=300) num_passes=300)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册