diff --git a/config.py b/configs/config.py similarity index 96% rename from config.py rename to configs/config.py index 0261b02..a539478 100644 --- a/config.py +++ b/configs/config.py @@ -7,11 +7,11 @@ from multiprocessing import cpu_count def use_fp32_config(): for config_file in [ - "32k.json", - "40k.json", - "48k.json", - "48k_v2.json", - "32k_v2.json", + "v1/32k.json", + "v1/40k.json", + "v1/48k.json", + "v2/48k.json", + "v2/32k.json", ]: with open(f"configs/{config_file}", "r") as f: strr = f.read().replace("true", "false") @@ -148,7 +148,7 @@ class Config: x_max = 32 if self.dml: print("use DirectML instead") - if(os.path.exists("runtime\Lib\site-packages\onnxruntime\capi\DirectML.dll")==False): + if os.path.exists("runtime\Lib\site-packages\onnxruntime\capi\DirectML.dll")==False: try: os.rename("runtime\Lib\site-packages\onnxruntime", "runtime\Lib\site-packages\onnxruntime-cuda") except: diff --git a/configs/32k.json b/configs/v1/32k.json similarity index 97% rename from configs/32k.json rename to configs/v1/32k.json index d5f16d6..400b6be 100644 --- a/configs/32k.json +++ b/configs/v1/32k.json @@ -7,7 +7,7 @@ "betas": [0.8, 0.99], "eps": 1e-9, "batch_size": 4, - "fp16_run": true, + "fp16_run": false, "lr_decay": 0.999875, "segment_size": 12800, "init_lr_ratio": 1, diff --git a/configs/40k.json b/configs/v1/40k.json similarity index 97% rename from configs/40k.json rename to configs/v1/40k.json index 4ffc87b..cb30b8b 100644 --- a/configs/40k.json +++ b/configs/v1/40k.json @@ -7,7 +7,7 @@ "betas": [0.8, 0.99], "eps": 1e-9, "batch_size": 4, - "fp16_run": true, + "fp16_run": false, "lr_decay": 0.999875, "segment_size": 12800, "init_lr_ratio": 1, diff --git a/configs/48k.json b/configs/v1/48k.json similarity index 97% rename from configs/48k.json rename to configs/v1/48k.json index 2d0e05b..6875991 100644 --- a/configs/48k.json +++ b/configs/v1/48k.json @@ -7,7 +7,7 @@ "betas": [0.8, 0.99], "eps": 1e-9, "batch_size": 4, - "fp16_run": true, + "fp16_run": false, "lr_decay": 0.999875, "segment_size": 11520, "init_lr_ratio": 1, diff --git a/configs/32k_v2.json b/configs/v2/32k.json similarity index 97% rename from configs/32k_v2.json rename to configs/v2/32k.json index 70e534f..36adb8a 100644 --- a/configs/32k_v2.json +++ b/configs/v2/32k.json @@ -7,7 +7,7 @@ "betas": [0.8, 0.99], "eps": 1e-9, "batch_size": 4, - "fp16_run": true, + "fp16_run": false, "lr_decay": 0.999875, "segment_size": 12800, "init_lr_ratio": 1, diff --git a/configs/48k_v2.json b/configs/v2/48k.json similarity index 97% rename from configs/48k_v2.json rename to configs/v2/48k.json index 75f770c..73ee363 100644 --- a/configs/48k_v2.json +++ b/configs/v2/48k.json @@ -7,7 +7,7 @@ "betas": [0.8, 0.99], "eps": 1e-9, "batch_size": 4, - "fp16_run": true, + "fp16_run": false, "lr_decay": 0.999875, "segment_size": 17280, "init_lr_ratio": 1,