diff --git a/vall_e/config.py b/vall_e/config.py index 76bc03e..42e4cf0 100644 --- a/vall_e/config.py +++ b/vall_e/config.py @@ -33,10 +33,10 @@ class Config(ConfigBase): warmup_max_lr: float = 2e-4 dis_warmup_max_lr: float = 4e-4 warmup_num_steps: int = 1_000 - max_iter: int = 10_000 + max_iter: int = 1_000_000 gradient_clipping: float = 100 eval_every: int = 2_000 - save_ckpt_every: int = 10_000 + save_ckpt_every: int = 2_000 model: str = "ar-quarter" spkr_name_getter: str = "lambda p: p.parts[-2]"