rename zoo names

lijp22 · Aug 10, 2021 · 9120e3a · 9120e3a
1 parent ab2b1e9
commit 9120e3a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/leaderboard_runner b/leaderboard_runner
@@ -18,10 +18,10 @@ if __name__ == '__main__':
 
     utils_dir = os.path.abspath('utils')
 
-    remote_model_zoo = 'oss://speechio-leaderboard/model-zoo'
+    remote_model_zoo = 'oss://speechio-leaderboard/models'
     model_zoo = os.path.abspath('models')
 
-    remote_dataset_zoo = 'oss://speechio-leaderboard/dataset-zoo'
+    remote_dataset_zoo = 'oss://speechio-leaderboard/datasets'
     dataset_zoo = os.path.abspath('datasets')
 
     model = ''

diff --git a/leaderboard_submit b/leaderboard_submit
@@ -10,6 +10,9 @@ if __name__ == '__main__':
     args = parser.parse_args()
     print(args, file=sys.stderr)
 
+    remote_model_zoo = 'oss://speechio-leaderboard/models'
+    remote_dataset_zoo = 'oss://speechio-leaderboard/datasets'
+
     if args.type == 'model':
         model_info_file = os.path.join(args.model_dir, 'model.yaml')
         assert(os.path.isfile(model_info_file))
@@ -32,7 +35,13 @@ if __name__ == '__main__':
         sys.exit(-1)
 
     src = args.model_dir
-    dst = os.path.join(F'oss://speechio-leaderboard/{args.type}-zoo', args.model_id) + '/'
+    if args.type == 'model':
+        dst = os.path.join(remote_model_zoo, args.model_id) + '/'
+    elif args.type == 'dataset':
+        raise NotImplementedError
+    else:
+        raise NotImplementedError
+
     cmd = F'./oss -c SAFEBOX/oss.cfg cp -ur {src} {dst}'
     print(cmd, file=sys.stderr)
     os.system(cmd)