mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-14 12:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor updates to the TTS & CODEC recipes

Browse Source
This commit is contained in:
JinZr 2024-10-21 17:17:59 +08:00
parent 8da9acd7e1
commit caa1d41b22
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
egs/libritts/CODEC/encodec/train.py
View File

@ -138,7 +138,7 @@ def get_parser():
parser.add_argument( parser.add_argument(
"--save-every-n", "--save-every-n",
type=int, type=int,
default=1, default=5,
help="""Save checkpoint after processing this number of epochs" help="""Save checkpoint after processing this number of epochs"
periodically. We save checkpoint to exp-dir/ whenever periodically. We save checkpoint to exp-dir/ whenever
params.cur_epoch % save_every_n == 0. The checkpoint filename params.cur_epoch % save_every_n == 0. The checkpoint filename
@ -1093,14 +1093,14 @@ def run(rank, world_size, args):
rank=rank, rank=rank,
) )
# if not params.print_diagnostics: if not params.print_diagnostics:
# scan_pessimistic_batches_for_oom( scan_pessimistic_batches_for_oom(
# model=model, model=model,
# train_dl=train_dl, train_dl=train_dl,
# optimizer_g=optimizer_g, optimizer_g=optimizer_g,
# optimizer_d=optimizer_d, optimizer_d=optimizer_d,
# params=params, params=params,
# ) )
scaler = GradScaler(enabled=params.use_fp16, init_scale=1.0) scaler = GradScaler(enabled=params.use_fp16, init_scale=1.0)
if checkpoints and "grad_scaler" in checkpoints: if checkpoints and "grad_scaler" in checkpoints:

2
egs/libritts/TTS/vits/train.py
View File

@ -212,7 +212,7 @@ def get_params() -> AttributeDict:
"log_interval": 50, "log_interval": 50,
"valid_interval": 200, "valid_interval": 200,
"env_info": get_env_info(), "env_info": get_env_info(),
"sampling_rate": 22050, "sampling_rate": 24000,
"frame_shift": 256, "frame_shift": 256,
"frame_length": 1024, "frame_length": 1024,
"feature_dim": 513, # 1024 // 2 + 1, 1024 is fft_length "feature_dim": 513, # 1024 // 2 + 1, 1024 is fft_length