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

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

@@ -19,6 +19,7 @@ import warnings
 import logging
 import paddle
 import sys
+import json
 from utils.oss_client import HangZhouOSSClient
 import utils.compress as compress
 
@@ -223,6 +224,9 @@ def main(args):
             fetch_var_names = config.get(
                 "runner.save_inference_fetch_varnames", [])
             fetchvars = []
+            loig.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: