提交 94a13bc6 编写于 作者: D dong zhihong

remove buffered function call"

上级 2113a409
...@@ -37,8 +37,7 @@ def train_lambdaRank(num_passes): ...@@ -37,8 +37,7 @@ def train_lambdaRank(num_passes):
paddle.dataset.mq2007.test, format="listwise") paddle.dataset.mq2007.test, format="listwise")
train_reader = paddle.batch( train_reader = paddle.batch(
paddle.reader.shuffle(fill_default_train, buf_size=100), batch_size=32) paddle.reader.shuffle(fill_default_train, buf_size=100), batch_size=32)
test_reader = paddle.batch( test_reader = paddle.batch(fill_default_test, batch_size=32)
paddle.reader.buffered(fill_default_test, size=100), batch_size=32)
# mq2007 input_dim = 46, dense format # mq2007 input_dim = 46, dense format
input_dim = 46 input_dim = 46
......
...@@ -54,9 +54,7 @@ def train_ranknet(num_passes): ...@@ -54,9 +54,7 @@ def train_ranknet(num_passes):
train_reader = paddle.batch( train_reader = paddle.batch(
paddle.reader.shuffle(paddle.dataset.mq2007.train, buf_size=100), paddle.reader.shuffle(paddle.dataset.mq2007.train, buf_size=100),
batch_size=100) batch_size=100)
test_reader = paddle.batch( test_reader = paddle.batch(paddle.dataset.mq2007.test, batch_size=100)
paddle.reader.buffered(paddle.dataset.mq2007.test, size=100),
batch_size=100)
# mq2007 feature_dim = 46, dense format # mq2007 feature_dim = 46, dense format
# fc hidden_dim = 128 # fc hidden_dim = 128
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册