丁云鹏 5 ヶ月 前
コミット
8116a955e6

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

@@ -244,7 +244,7 @@ class Main(object):
 
         if fleet.is_first_worker():
             model_dir = "{}/{}".format(save_model_path, epochs)
-            oss_object_name = self.config.get("runner.oss_object_name")
+            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每个参数一个文件,需要同这种方式加载
             program, feed_target_names, fetch_targets = paddle.static.load_inference_model(
@@ -274,9 +274,9 @@ class Main(object):
             compress.compress_tar(os.path.join(model_dir, "dnn_plugin_new"), "dnn_plugin_new.tar.gz")
             client = HangZhouOSSClient("art-recommend")
             client.put_object_from_file(oss_object_name, "dnn_plugin_new.tar.gz")
-            while True:
-                time.sleep(300)
-                continue;
+        while True:
+            time.sleep(300)
+            continue;
 
         if reader_type == "InmemoryDataset":
             self.reader.release_memory()