Add files via upload

2023-06-18 19:37:53 +08:00 · 2023-06-18 19:37:53 +08:00 · 0812020c90
parent 66667c8f50
commit 0812020c90
1 changed files with 4 additions and 1 deletions
--- a/train/utils.py
+++ b/train/utils.py
@ -360,7 +360,10 @@ def get_hparams(init=True):
    if not os.path.exists(experiment_dir):
        os.makedirs(experiment_dir)

-    config_path = "configs/%s.json" % args.sample_rate
+    if(args.version=="v1"or args.sample_rate=="40k"):
+        config_path = "configs/%s.json" % args.sample_rate
+    else:
+        config_path = "configs/%s_v2.json" % args.sample_rate
    config_save_path = os.path.join(experiment_dir, "config.json")
    if init:
        with open(config_path, "r") as f: