From 7e861f0ee4b2c23821be93c3483acf59cdd8e9a1 Mon Sep 17 00:00:00 2001 From: "yi.wu" Date: Thu, 7 Jun 2018 22:19:02 +0800 Subject: [PATCH] fix small bug in caculating speed --- benchmark/fluid/fluid_benchmark.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/benchmark/fluid/fluid_benchmark.py b/benchmark/fluid/fluid_benchmark.py index bd0243aa60..62a05234c4 100644 --- a/benchmark/fluid/fluid_benchmark.py +++ b/benchmark/fluid/fluid_benchmark.py @@ -371,8 +371,7 @@ def train_parallel(avg_loss, infer_prog, optimizer, train_reader, test_reader, print("Pass %d, batch %d, loss %s" % (pass_id, batch_id, np.array(loss))) batch_id += 1 - if args.use_reader_op: - num_samples = num_samples * args.gpus + print_train_time(start_time, time.time(), num_samples) if not args.no_test and batch_acc: test_acc = test(startup_exe, infer_prog, test_reader, feeder, -- GitLab