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

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

@@ -225,9 +225,7 @@ def main(args):
                 "runner.save_inference_fetch_varnames", [])
             fetchvars = []
 
-            for op in enumerate(paddle.static.default_main_program().global_block().ops):
-                for name in op.input_arg_names:
-                    logger.info("op.input_arg_names {}".format(name))
+            logger.info("global_block().to_string {}".format(paddle.static.default_main_program().global_block().to_string(throw_on_error=True)))
             
 
             for var_name in feed_var_names: