丁云鹏 5 月之前
父節點
當前提交
157b893948

+ 1 - 1
recommend-model-produce/src/main/python/models/dnn/config.yaml

@@ -35,7 +35,7 @@ runner:
   #use inference save model
   # model_init_path: "output_model_dnn/2" # init model
   use_inference: True
-  save_inference_feed_varnames: ["1","2","3","4","5","6","7","8","9","10","11","12","13","14","15","16","17","18","19","20","21","22","23","24","25","26","dense_input"]
+  save_inference_feed_varnames: ["dense_input"]
   save_inference_fetch_varnames: ["label", "SparseFeatFactors"]
 
   # distribute_config

+ 0 - 3
recommend-model-produce/src/main/python/tools/static_trainer.py

@@ -224,9 +224,6 @@ def main(args):
             fetch_var_names = config.get(
                 "runner.save_inference_fetch_varnames", [])
             fetchvars = []
-            logger.info("Paddle.static.default_main_program().global_block().ops: {}"
-                        .format(json.dumps(paddle.static.default_main_program()
-                               .global_block().ops)))
             for var_name in feed_var_names:
                 if var_name not in paddle.static.default_main_program(
                 ).global_block().vars: