Skip to content

Commit 3ba7e74

Browse files
committed
use clone(for_test=True) replace get_inference_program
1 parent 87086b1 commit 3ba7e74

File tree

3 files changed

+2
-23
lines changed

3 files changed

+2
-23
lines changed

paddle/fluid/API.spec

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,6 @@ paddle.fluid.io.load_params ArgSpec(args=['executor', 'dirname', 'main_program',
7373
paddle.fluid.io.load_persistables ArgSpec(args=['executor', 'dirname', 'main_program', 'filename'], varargs=None, keywords=None, defaults=(None, None))
7474
paddle.fluid.io.save_inference_model ArgSpec(args=['dirname', 'feeded_var_names', 'target_vars', 'executor', 'main_program', 'model_filename', 'params_filename', 'export_for_deployment'], varargs=None, keywords=None, defaults=(None, None, None, True))
7575
paddle.fluid.io.load_inference_model ArgSpec(args=['dirname', 'executor', 'model_filename', 'params_filename', 'pserver_endpoints'], varargs=None, keywords=None, defaults=(None, None, None))
76-
paddle.fluid.io.get_inference_program ArgSpec(args=['target_vars', 'main_program'], varargs=None, keywords=None, defaults=(None,))
7776
paddle.fluid.initializer.ConstantInitializer.__init__ ArgSpec(args=['self', 'value', 'force_cpu'], varargs=None, keywords=None, defaults=(0.0, False))
7877
paddle.fluid.initializer.UniformInitializer.__init__ ArgSpec(args=['self', 'low', 'high', 'seed'], varargs=None, keywords=None, defaults=(-1.0, 1.0, 0))
7978
paddle.fluid.initializer.NormalInitializer.__init__ ArgSpec(args=['self', 'loc', 'scale', 'seed'], varargs=None, keywords=None, defaults=(0.0, 1.0, 0))

python/paddle/fluid/io.py

Lines changed: 1 addition & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,7 @@
2727

2828
__all__ = [
2929
'save_vars', 'save_params', 'save_persistables', 'load_vars', 'load_params',
30-
'load_persistables', 'save_inference_model', 'load_inference_model',
31-
'get_inference_program'
30+
'load_persistables', 'save_inference_model', 'load_inference_model'
3231
]
3332

3433

@@ -504,23 +503,6 @@ def load_persistables(executor, dirname, main_program=None, filename=None):
504503
filename=filename)
505504

506505

507-
def get_inference_program(target_vars, main_program=None):
508-
if main_program is None:
509-
main_program = default_main_program()
510-
if not isinstance(target_vars, list):
511-
target_vars = [target_vars]
512-
vars = []
513-
for var in target_vars:
514-
if isinstance(var, Evaluator):
515-
vars.extend(var.states)
516-
vars.extend(var.metrics)
517-
else:
518-
vars.append(var)
519-
pruned_program = main_program._prune(targets=vars)
520-
inference_program = pruned_program._inference_optimize()
521-
return inference_program
522-
523-
524506
def prepend_feed_ops(inference_program,
525507
feed_target_names,
526508
feed_holder_name='feed'):

python/paddle/fluid/tests/unittests/dist_transformer.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -440,9 +440,7 @@ def split_data(data, num_part):
440440
def test_context(train_progm, avg_cost, train_exe, dev_count, data_input_names,
441441
sum_cost, token_num):
442442
# Context to do validation.
443-
test_program = train_progm.clone()
444-
with fluid.program_guard(test_program):
445-
test_program = fluid.io.get_inference_program([avg_cost])
443+
test_program = train_progm.clone(for_test=True)
446444

447445
val_data = DataReader(
448446
src_vocab_fpath=TrainTaskConfig.src_vocab_fpath,

0 commit comments

Comments
 (0)