소스 검색

dssm train

丁云鹏 4 달 전
부모
커밋
94b8953350
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      recommend-model-produce/src/main/python/tools/static_ps_infer_v2.py

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

@@ -235,7 +235,7 @@ class Main(object):
         model_mode = config.get("runner.model_mode", 0)
         client = HangZhouOSSClient("art-recommend")
         oss_object_name = self.config.get("runner.oss_object_name", "dyp/model.tar.gz")
-        client.get_object_to_file("oss_object_name", "model.tar.gz")
+        client.get_object_to_file(oss_object_name, "model.tar.gz")
         compress.uncompress_tar("model.tar.gz", init_model_path)
         assert os.path.exists(init_model_path)