丁云鹏 5 月之前
父节点
当前提交
6b1ec3d942
共有 1 个文件被更改,包括 5 次插入0 次删除
  1. 5 0
      recommend-model-produce/src/main/python/tools/static_trainer.py

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

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