丁云鹏 4 ماه پیش
والد
کامیت
7b7c1b1292

+ 1 - 0
.gitignore

@@ -1,3 +1,4 @@
+*.DS_Store
 # ---> Java
 *.class
 

+ 1 - 1
recommend-model-produce/src/main/python/models/dssm/config_ps_hdfs.yaml

@@ -32,7 +32,7 @@ runner:
   test_data_dir: "data/test"
   infer_reader_path: "bq_reader_infer"  # importlib format
   infer_batch_size: 1
-  infer_load_path: "output_model_dssm"
+  infer_load_path: "/app/output_model_dssm"
   infer_start_epoch: 0
   infer_end_epoch: 1
 

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

@@ -159,6 +159,7 @@ class Main(object):
         client = HangZhouOSSClient("art-recommend")
         client.get_object_to_file("dyp/test.tar.gz", "test.tar.gz")
         compress.uncompress_tar("test.tar.gz", init_model_path)
+        assert os.path.exists(init_model_path)
 
         #if fleet.is_first_worker():
         #fleet.load_inference_model(init_model_path, mode=int(model_mode))