Create models/config.yaml
Browse files- models/config.yaml +39 -0
models/config.yaml
ADDED
|
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
data:
|
| 2 |
+
block_size: 512
|
| 3 |
+
duration: 2
|
| 4 |
+
encoder: hubertsoft
|
| 5 |
+
encoder_ckpt: pretrain/hubert/hubert-soft-0d54a1f4.pt
|
| 6 |
+
encoder_hop_size: 320
|
| 7 |
+
encoder_out_channels: 256
|
| 8 |
+
encoder_sample_rate: 16000
|
| 9 |
+
f0_extractor: crepe
|
| 10 |
+
f0_max: 800
|
| 11 |
+
f0_min: 65
|
| 12 |
+
sampling_rate: 44100
|
| 13 |
+
train_path: data/train
|
| 14 |
+
valid_path: data/val
|
| 15 |
+
device: cuda
|
| 16 |
+
enhancer:
|
| 17 |
+
ckpt: pretrain/nsf_hifigan/model
|
| 18 |
+
type: nsf-hifigan
|
| 19 |
+
env:
|
| 20 |
+
expdir: exp/combsub
|
| 21 |
+
gpu_id: 0
|
| 22 |
+
loss:
|
| 23 |
+
fft_max: 2048
|
| 24 |
+
fft_min: 256
|
| 25 |
+
n_scale: 4
|
| 26 |
+
model:
|
| 27 |
+
n_spk: 1
|
| 28 |
+
type: CombSubFast
|
| 29 |
+
train:
|
| 30 |
+
batch_size: 24
|
| 31 |
+
cache_all_data: true
|
| 32 |
+
cache_device: cuda
|
| 33 |
+
cache_fp16: true
|
| 34 |
+
epochs: 100000
|
| 35 |
+
interval_log: 10
|
| 36 |
+
interval_val: 2000
|
| 37 |
+
lr: 0.0005
|
| 38 |
+
num_workers: 2
|
| 39 |
+
weight_decay: 0
|