丁云鹏 6 months ago
parent
commit
b56c09706b

+ 3 - 3
recommend-model-produce/src/main/python/models/dnn/config.yaml

@@ -42,9 +42,9 @@ runner:
   sync_mode: "async"
   split_file_list: False
   thread_num: 1
-  upload_oss: true
-  oss_filename: dnn.tar.gz
-  oss_path: /dyp
+
+  upload_oss: True
+  oss_object_name: "/dyp/dnn.tar.gz"
 
 
 # hyper parameters of user-defined network

+ 2 - 2
recommend-model-produce/src/main/python/tools/static_trainer.py

@@ -102,9 +102,9 @@ def main(args):
 
     logger.info("**************common.configs**********")
     logger.info(
-        "use_gpu: {}, use_xpu: {}, use_visual: {}, train_batch_size: {}, train_data_dir: {}, epochs: {}, print_interval: {}, model_save_path: {}".
+        "use_gpu: {}, use_xpu: {}, use_visual: {}, train_batch_size: {}, train_data_dir: {}, epochs: {}, print_interval: {}, model_save_path: {}, oss_object_name {}".
         format(use_gpu, use_xpu, use_visual, batch_size, train_data_dir,
-               epochs, print_interval, model_save_path))
+               epochs, print_interval, model_save_path, oss_object_name))
     logger.info("**************common.configs**********")
 
     if use_xpu: