소스 검색

save model

丁云鹏 5 달 전
부모
커밋
3b34a46d9c
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

@@ -243,7 +243,7 @@ class Main(object):
                     [self.inference_target_var], self.exe)
 
         if fleet.is_first_worker():
-            model_dir = "{}/{}".format(save_model_path, epochs)
+            model_dir = "{}/{}".format(save_model_path, epochs - 1)
             oss_object_name = self.config.get("runner.oss_object_name", "dyp/dnn_plugin_new.tar.gz")
             # trans to new format
             # {"model_filename":"", "params_filename":""} fleet每个参数一个文件,需要同这种方式加载