Spaces:
Running
Running
JacobLinCool
commited on
Commit
•
771996d
1
Parent(s):
38548f2
perf: set batch size to 32
Browse files- config.json +3 -3
- infer/modules/train/train.py +1 -1
config.json
CHANGED
@@ -58,7 +58,7 @@
|
|
58 |
"use_spectral_norm": false
|
59 |
},
|
60 |
"train": {
|
61 |
-
"batch_size":
|
62 |
"betas": [
|
63 |
0.8,
|
64 |
0.99
|
@@ -67,7 +67,7 @@
|
|
67 |
"c_mel": 45,
|
68 |
"epochs": 20000,
|
69 |
"eps": 1e-09,
|
70 |
-
"fp16_run":
|
71 |
"init_lr_ratio": 1,
|
72 |
"learning_rate": 0.0001,
|
73 |
"log_interval": 200,
|
@@ -76,4 +76,4 @@
|
|
76 |
"segment_size": 12800,
|
77 |
"warmup_epochs": 0
|
78 |
}
|
79 |
-
}
|
|
|
58 |
"use_spectral_norm": false
|
59 |
},
|
60 |
"train": {
|
61 |
+
"batch_size": 32,
|
62 |
"betas": [
|
63 |
0.8,
|
64 |
0.99
|
|
|
67 |
"c_mel": 45,
|
68 |
"epochs": 20000,
|
69 |
"eps": 1e-09,
|
70 |
+
"fp16_run": false,
|
71 |
"init_lr_ratio": 1,
|
72 |
"learning_rate": 0.0001,
|
73 |
"log_interval": 200,
|
|
|
76 |
"segment_size": 12800,
|
77 |
"warmup_epochs": 0
|
78 |
}
|
79 |
+
}
|
infer/modules/train/train.py
CHANGED
@@ -98,7 +98,7 @@ def train(exp_dir: str):
|
|
98 |
hps.pretrainD = "assets/pretrained_v2/f0D40k.pth"
|
99 |
hps.version = "v2"
|
100 |
hps.gpus = "0"
|
101 |
-
hps.train.batch_size =
|
102 |
hps.sample_rate = "40k"
|
103 |
hps.if_f0 = 1
|
104 |
hps.if_latest = 0
|
|
|
98 |
hps.pretrainD = "assets/pretrained_v2/f0D40k.pth"
|
99 |
hps.version = "v2"
|
100 |
hps.gpus = "0"
|
101 |
+
hps.train.batch_size = 32
|
102 |
hps.sample_rate = "40k"
|
103 |
hps.if_f0 = 1
|
104 |
hps.if_latest = 0
|