Fix: add use_jp_extra parameter to config.json (for non-jp-extra training)

This commit is contained in:
litagin02
2024-03-11 17:51:55 +09:00
parent 44851a6219
commit b3cc705caa
2 changed files with 3 additions and 2 deletions

View File

@@ -24,6 +24,7 @@
"freeze_encoder": false "freeze_encoder": false
}, },
"data": { "data": {
"use_jp_extra": false,
"training_files": "Data/your_model_name/filelists/train.list", "training_files": "Data/your_model_name/filelists/train.list",
"validation_files": "Data/your_model_name/filelists/val.list", "validation_files": "Data/your_model_name/filelists/val.list",
"max_wav_value": 32768.0, "max_wav_value": 32768.0,

View File

@@ -16,7 +16,7 @@ class HyperParametersTrain(BaseModel):
seed: int = 42 seed: int = 42
epochs: int = 1000 epochs: int = 1000
learning_rate: float = 0.0001 learning_rate: float = 0.0001
betas: list[float] = [0.8, 0.99] betas: tuple[float, float] = (0.8, 0.99)
eps: float = 1e-9 eps: float = 1e-9
batch_size: int = 2 batch_size: int = 2
bf16_run: bool = False bf16_run: bool = False
@@ -50,7 +50,7 @@ class HyperParametersData(BaseModel):
mel_fmin: float = 0.0 mel_fmin: float = 0.0
mel_fmax: Optional[float] = None mel_fmax: Optional[float] = None
add_blank: bool = True add_blank: bool = True
n_speakers: int = 512 n_speakers: int = 1
cleaned_text: bool = True cleaned_text: bool = True
spk2id: dict[str, int] = { spk2id: dict[str, int] = {
"Dummy": 0, "Dummy": 0,