update

examples/vm_sound_classification/step_3_train_model.py

@@ -162,7 +162,7 @@ def main():
     )
 
     # models
-    logger.info("prepare models")
+    logger.info(f"prepare models. config_file: {args.config_file}")
     config = CnnAudioClassifierConfig.from_pretrained(
         pretrained_model_name_or_path=args.config_file,
         # num_labels=vocabulary.get_vocab_size(namespace="labels")

main.py

@@ -26,22 +26,27 @@ def get_args():
         default=(project_path / "data/examples").as_posix(),
         type=str
     )
+    parser.add_argument(
+        "--models_repo_id",
+        default="qgyd2021/vm_sound_classification",
+        type=str
+    )
     parser.add_argument(
         "--trained_model_dir",
         default=(project_path / "trained_models").as_posix(),
         type=str
     )
+    parser.add_argument(
+        "--hf_token",
+        default=environment.get("hf_token"),
+        type=str,
+    )
     parser.add_argument(
         "--server_port",
         default=environment.get("server_port", 7860),
         type=int
     )
 
-    parser.add_argument(
-        "--models_repo_id",
-        default="qgyd2021/vm_sound_classification",
-        type=str
-    )
     args = parser.parse_args()
     return args
 
@@ -118,7 +123,8 @@ def main():
         trained_model_dir.mkdir(parents=True, exist_ok=True)
         _ = snapshot_download(
             repo_id=args.models_repo_id,
-            local_dir=trained_model_dir.as_posix()
+            local_dir=trained_model_dir.as_posix(),
+            token=args.hf_token,
         )
 
     # examples