fix code style

wangkuiyi-patch-1
guochaorong 7 years ago
parent a95cf55eeb
commit 90a74a9f2f

@ -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

Loading…
Cancel
Save