Spaces:
Paused
Paused
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
|