丁云鹏 5 months ago
parent
commit
2682c4cc42
1 changed files with 3 additions and 1 deletions
  1. 3 1
      recommend-model-produce/src/main/python/tools/static_trainer.py

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

@@ -225,7 +225,9 @@ def main(args):
                 "runner.save_inference_fetch_varnames", [])
             fetchvars = []
 
-            logger.info("global_block().to_string {}".format(paddle.static.default_main_program().global_block().to_string(throw_on_error=True)))
+            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))
             
 
             for var_name in feed_var_names: