Parcourir la source

Change config path

Ruslan Mavlyutov il y a 2 ans
Parent
commit
79546ff4e6
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      scripts/m4t/finetune/dataset.py

+ 1 - 1
scripts/m4t/finetune/dataset.py

@@ -168,7 +168,7 @@ def main(args: Namespace) -> None:
     manifest_path = download_fleurs_dataset(
         source_lang=args.source_lang,
         target_lang=args.target_lang,
-        unit_extractor_config="/checkpoint/krs/unit_extraction/xlsr1b/lang41_10k_xlsr_lyr35.yaml",
+        unit_extractor_config="lang41_10k_xlsr_lyr35.yaml",
         split=args.split,
         save_directory=args.save_dir,
     )