diff --git a/train/utils.py b/train/utils.py
index 434ed7a..984c8c1 100644
--- 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: