hugo flores garcia commited on
Commit
f4d4a42
·
1 Parent(s): 3df717c

better fine tune output

Browse files
Files changed (1) hide show
  1. scripts/exp/fine_tune.py +8 -2
scripts/exp/fine_tune.py CHANGED
@@ -68,8 +68,14 @@ def fine_tune(audio_files_or_folders: List[str], name: str):
68
  yaml.dump(interface_conf, f)
69
 
70
 
71
- print(f"generated confs in {finetune_dir}. run training jobs with `python scripts/exp/train.py --args.load {finetune_dir}/<c2f/coarse>.yml` ")
72
-
 
 
 
 
 
 
73
  if __name__ == "__main__":
74
  args = argbind.parse_args()
75
 
 
68
  yaml.dump(interface_conf, f)
69
 
70
 
71
+ # print(f"generated confs in {finetune_dir}.
72
+ # run training jobs with `python scripts/exp/train.py --args.load {finetune_dir}/<c2f/coarse>.yml` ")
73
+
74
+ print(f"generated confs in {finetune_dir}.")
75
+ print()
76
+ print(f"you'll need to run two training jobs, though they can run in parallel on separate GPUs.")
77
+ print(f"run the coarse job with \n\tpython scripts/exp/train.py --args.load {finetune_dir}/coarse.yml\n")
78
+ print(f"run the c2f job with \n\tpython scripts/exp/train.py --args.load {finetune_dir}/c2f.yml\n")
79
  if __name__ == "__main__":
80
  args = argbind.parse_args()
81