浏览代码

save model

丁云鹏 5 月之前
父节点
当前提交
c615280b99
共有 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

@@ -260,7 +260,7 @@ class Main(object):
                 self.exe,
                 self.exe,
                 program = program)
                 program = program)
 
 
-            # logger.info("program.global_block().vars.keys() {}".format(program.global_block().vars.keys()))
+            logger.info("program.global_block().vars.keys() {}".format(program.global_block().vars.keys()))
 
 
             compress.compress_tar(os.path.join(model_dir, "dnn_plugin_new"), "dnn_plugin_new.tar.gz")
             compress.compress_tar(os.path.join(model_dir, "dnn_plugin_new"), "dnn_plugin_new.tar.gz")
             client = HangZhouOSSClient("art-recommend")
             client = HangZhouOSSClient("art-recommend")