提交 90a74a9f 编写于 作者: G guochaorong

fix code style

上级 a95cf55e
...@@ -201,8 +201,8 @@ def get_model(args): ...@@ -201,8 +201,8 @@ def get_model(args):
train_reader, buf_size=5120), train_reader, buf_size=5120),
batch_size=args.batch_size * args.gpus, batch_size=args.batch_size * args.gpus,
drop_last=True) drop_last=True)
batched_test_reader = paddle.batch(train_reader, batched_test_reader = paddle.batch(
batch_size=args.batch_size, drop_last=True) train_reader, batch_size=args.batch_size, drop_last=True)
return avg_cost, inference_program, optimizer, batched_train_reader,\ return avg_cost, inference_program, optimizer, batched_train_reader,\
batched_test_reader, batch_acc batched_test_reader, batch_acc
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册