Skip to content

Commit 8fa457f

Browse files
author
chengduo
authored
Merge pull request #11296 from typhoonzero/fix_fluid_benchmark_performance_data
fix small bug in caculating speed
2 parents 80e882a + 7e861f0 commit 8fa457f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

benchmark/fluid/fluid_benchmark.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -371,8 +371,7 @@ def train_parallel(avg_loss, infer_prog, optimizer, train_reader, test_reader,
371371
print("Pass %d, batch %d, loss %s" %
372372
(pass_id, batch_id, np.array(loss)))
373373
batch_id += 1
374-
if args.use_reader_op:
375-
num_samples = num_samples * args.gpus
374+
376375
print_train_time(start_time, time.time(), num_samples)
377376
if not args.no_test and batch_acc:
378377
test_acc = test(startup_exe, infer_prog, test_reader, feeder,

0 commit comments

Comments
 (0)