|
@@ -266,14 +266,6 @@ class Main(object):
|
|
program.global_block().vars.keys()))
|
|
program.global_block().vars.keys()))
|
|
else:
|
|
else:
|
|
feedvars.append(program.global_block().vars[var_name])
|
|
feedvars.append(program.global_block().vars[var_name])
|
|
- for var_name in fetch_targets:
|
|
|
|
- if var_name not in program.global_block().vars:
|
|
|
|
- raise ValueError(
|
|
|
|
- "Fetch variable: {} not in default_main_program, global block has follow vars: {}".
|
|
|
|
- format(var_name,
|
|
|
|
- program.global_block().vars.keys()))
|
|
|
|
- else:
|
|
|
|
- fetchvars.append(program.global_block().vars[var_name])
|
|
|
|
|
|
|
|
paddle.static.save_inference_model(
|
|
paddle.static.save_inference_model(
|
|
os.path.join(model_dir, "dnn_plugin_new"),
|
|
os.path.join(model_dir, "dnn_plugin_new"),
|