1
0
mirror of synced 2024-12-18 02:16:08 +01:00

replace configs

This commit is contained in:
Ftps 2023-08-19 19:36:15 +09:00
parent b562b12cb9
commit 78dae1dd09
6 changed files with 11 additions and 11 deletions

View File

@ -7,11 +7,11 @@ from multiprocessing import cpu_count
def use_fp32_config(): def use_fp32_config():
for config_file in [ for config_file in [
"32k.json", "v1/32k.json",
"40k.json", "v1/40k.json",
"48k.json", "v1/48k.json",
"48k_v2.json", "v2/48k.json",
"32k_v2.json", "v2/32k.json",
]: ]:
with open(f"configs/{config_file}", "r") as f: with open(f"configs/{config_file}", "r") as f:
strr = f.read().replace("true", "false") strr = f.read().replace("true", "false")
@ -148,7 +148,7 @@ class Config:
x_max = 32 x_max = 32
if self.dml: if self.dml:
print("use DirectML instead") 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: try:
os.rename("runtime\Lib\site-packages\onnxruntime", "runtime\Lib\site-packages\onnxruntime-cuda") os.rename("runtime\Lib\site-packages\onnxruntime", "runtime\Lib\site-packages\onnxruntime-cuda")
except: except:

View File

@ -7,7 +7,7 @@
"betas": [0.8, 0.99], "betas": [0.8, 0.99],
"eps": 1e-9, "eps": 1e-9,
"batch_size": 4, "batch_size": 4,
"fp16_run": true, "fp16_run": false,
"lr_decay": 0.999875, "lr_decay": 0.999875,
"segment_size": 12800, "segment_size": 12800,
"init_lr_ratio": 1, "init_lr_ratio": 1,

View File

@ -7,7 +7,7 @@
"betas": [0.8, 0.99], "betas": [0.8, 0.99],
"eps": 1e-9, "eps": 1e-9,
"batch_size": 4, "batch_size": 4,
"fp16_run": true, "fp16_run": false,
"lr_decay": 0.999875, "lr_decay": 0.999875,
"segment_size": 12800, "segment_size": 12800,
"init_lr_ratio": 1, "init_lr_ratio": 1,

View File

@ -7,7 +7,7 @@
"betas": [0.8, 0.99], "betas": [0.8, 0.99],
"eps": 1e-9, "eps": 1e-9,
"batch_size": 4, "batch_size": 4,
"fp16_run": true, "fp16_run": false,
"lr_decay": 0.999875, "lr_decay": 0.999875,
"segment_size": 11520, "segment_size": 11520,
"init_lr_ratio": 1, "init_lr_ratio": 1,

View File

@ -7,7 +7,7 @@
"betas": [0.8, 0.99], "betas": [0.8, 0.99],
"eps": 1e-9, "eps": 1e-9,
"batch_size": 4, "batch_size": 4,
"fp16_run": true, "fp16_run": false,
"lr_decay": 0.999875, "lr_decay": 0.999875,
"segment_size": 12800, "segment_size": 12800,
"init_lr_ratio": 1, "init_lr_ratio": 1,

View File

@ -7,7 +7,7 @@
"betas": [0.8, 0.99], "betas": [0.8, 0.99],
"eps": 1e-9, "eps": 1e-9,
"batch_size": 4, "batch_size": 4,
"fp16_run": true, "fp16_run": false,
"lr_decay": 0.999875, "lr_decay": 0.999875,
"segment_size": 17280, "segment_size": 17280,
"init_lr_ratio": 1, "init_lr_ratio": 1,