丁云鹏 5 місяців тому
батько
коміт
6b1ec3d942

+ 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: