Skip to content

Commit 38143e5

Browse files
committed
Clean unused changes
test=develop
1 parent d424115 commit 38143e5

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

benchmark/fluid/fluid_benchmark.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def train_parallel(train_args, test_args, args, train_prog, test_prog,
168168
startup_exe = fluid.Executor(place)
169169
startup_exe.run(startup_prog)
170170
strategy = fluid.ExecutionStrategy()
171-
strategy.num_threads = 0 #args.cpus
171+
strategy.num_threads = args.cpus
172172
strategy.allow_op_delay = False
173173
build_strategy = fluid.BuildStrategy()
174174
if args.reduce_strategy == "reduce":
@@ -188,8 +188,6 @@ def train_parallel(train_args, test_args, args, train_prog, test_prog,
188188
num_trainers = 1
189189
trainer_id = 0
190190

191-
print('Use parallel_executor')
192-
strategy.type = 2
193191
exe = fluid.ParallelExecutor(
194192
True,
195193
avg_loss.name,

benchmark/fluid/models/resnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,7 @@ def get_model(args, is_train, main_prog, startup_prog):
172172
reader, dshape, class_dim = _model_reader_dshape_classdim(args, is_train)
173173

174174
pyreader = None
175-
trainer_count = int(os.getenv("PADDLE_TRAINERS", 1))
175+
trainer_count = int(os.getenv("PADDLE_TRAINERS"))
176176
with fluid.program_guard(main_prog, startup_prog):
177177
with fluid.unique_name.guard():
178178
if args.use_reader_op:

0 commit comments

Comments
 (0)