replace configs
This commit is contained in:
parent
b562b12cb9
commit
78dae1dd09
@ -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:
|
@ -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,
|
@ -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,
|
@ -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,
|
@ -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,
|
@ -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,
|
Loading…
Reference in New Issue
Block a user