From cd330578ef0179c6c302c7fc9fd260847d28f005 Mon Sep 17 00:00:00 2001 From: "yi.wu" Date: Thu, 7 Jun 2018 09:26:01 +0800 Subject: [PATCH] add num_passes --- benchmark/fluid/models/mnist.py | 3 ++- benchmark/fluid/models/resnet.py | 3 ++- benchmark/fluid/models/vgg.py | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/benchmark/fluid/models/mnist.py b/benchmark/fluid/models/mnist.py index d903a834ec0..8e740dc6896 100644 --- a/benchmark/fluid/models/mnist.py +++ b/benchmark/fluid/models/mnist.py @@ -75,7 +75,8 @@ def get_model(args): shapes=[[-1, 1, 28, 28], (-1, 1)], lod_levels=[0, 0], dtypes=["float32", "int64"], - thread_num=args.gpus) + thread_num=args.gpus, + pass_num=args.pass_num) data_file = fluid.layers.double_buffer( fluid.layers.batch( data_file, batch_size=args.batch_size)) diff --git a/benchmark/fluid/models/resnet.py b/benchmark/fluid/models/resnet.py index 88451064fcc..2ee2b5be09b 100644 --- a/benchmark/fluid/models/resnet.py +++ b/benchmark/fluid/models/resnet.py @@ -157,7 +157,8 @@ def get_model(args): shapes=[[-1] + dshape, (-1, 1)], lod_levels=[0, 0], dtypes=["float32", "int64"], - thread_num=args.gpus) + thread_num=args.gpus, + pass_num=args.pass_num) data_file = fluid.layers.double_buffer( fluid.layers.batch( data_file, batch_size=args.batch_size)) diff --git a/benchmark/fluid/models/vgg.py b/benchmark/fluid/models/vgg.py index cb0dc977634..6092cdeb884 100644 --- a/benchmark/fluid/models/vgg.py +++ b/benchmark/fluid/models/vgg.py @@ -75,7 +75,8 @@ def get_model(args): shapes=[[-1] + data_shape, (-1, 1)], lod_levels=[0, 0], dtypes=["float32", "int64"], - thread_num=args.gpus) + thread_num=args.gpus, + pass_num=args.pass_num) data_file = fluid.layers.double_buffer( fluid.layers.batch( data_file, batch_size=args.batch_size)) -- GitLab