浏览代码

save model

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

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

@@ -256,7 +256,7 @@ class Main(object):
                 params_filename=None)
                 params_filename=None)
 
 
             feedvars = []
             feedvars = []
-            fetchvars = []
+            fetchvars = fetch_targets
 
 
             for var_name in feed_target_names:
             for var_name in feed_target_names:
                 if var_name not in program.global_block().vars:
                 if var_name not in program.global_block().vars: