dnn test

recommend-model-produce/src/main/python/tools/utils/save_load.py

@@ -79,7 +79,7 @@ def save_inference_model(model_path,
 
     program = _get_valid_program(None)
     copy_program = program.clone()
-    logger.info("program.clone(): {}".format(program.global_block().vars.keys()))
+    logger.info("program.clone(): {}".format(copy_program.global_block().vars.keys()))
 
     global_block = copy_program.global_block()
     remove_op_idx = []
@@ -104,7 +104,7 @@ def save_inference_model(model_path,
     for idx in remove_op_idx[::-1]:
         global_block._remove_op(idx)
     copy_program.desc.flush()
-    logger.info("copy_program.desc.flush(): {}".format(program.global_block().vars.keys()))
+    logger.info("copy_program.desc.flush(): {}".format(copy_program.global_block().vars.keys()))
 
     feed_var_names = [var.name for var in feed_vars]
 
@@ -113,9 +113,9 @@ def save_inference_model(model_path,
         copy_program = copy_program._prune_with_input(
             feeded_var_names=feed_var_names, targets=fetch_vars
         )
-        logger.info("copy_program._prune_with_input(): {}".format(program.global_block().vars.keys()))
+        logger.info("copy_program._prune_with_input(): {}".format(copy_program.global_block().vars.keys()))
     copy_program = copy_program._inference_optimize(prune_read_op=True)
-    logger.info("copy_program._inference_optimize(): {}".format(program.global_block().vars.keys()))
+    logger.info("copy_program._inference_optimize(): {}".format(copy_program.global_block().vars.keys()))