| train: |
| seed: 1234 |
| epochs: 100 |
| batch_size: 6 |
| gradient_accumulation: 4 |
| save_every_n_epoch: 1 |
| precision: 32 |
| gradient_clip: 1.0 |
| optimizer: |
| lr: 0.01 |
| lr_init: 0.00001 |
| lr_end: 0.0001 |
| warmup_steps: 2000 |
| decay_steps: 40000 |
| data: |
| max_eval_sample: 8 |
| max_sec: 40 |
| num_workers: 1 |
| pad_val: 1024 |
| model: |
| saving_path: "ckpt/" |
| resume_checkpoint: null |
| vocoder_config_path: "quantizer/new_ckpt/config.json" |
| vocoder_ckpt_path: "quantizer/new_ckpt/g_00600000" |
| datadir: "/home/liweiche/GigaSpeech/wavs" |
| metapath: "/home/liweiche/GigaSpeech/train2.json" |
| val_metapath: "/home/liweiche/GigaSpeech/dev2.json" |
| sampledir: "logs/" |
| pretrained_path: null |
| lr: 0.0001 |
| batch_size: 200.0 |
| train_bucket_size: 8192 |
| training_step: 800000 |
| optim_flat_percent: 0.0 |
| warmup_step: 50 |
| adam_beta1: 0.9 |
| adam_beta2: 0.98 |
| ffd_size: 3072 |
| hidden_size: 768 |
| enc_nlayers: 6 |
| dec_nlayers: 6 |
| nheads: 12 |
| ar_layer: 4 |
| ar_ffd_size: 1024 |
| ar_hidden_size: 256 |
| ar_nheads: 4 |
| aligner_softmax_temp: 1.0 |
| layer_norm_eps: 0.00001 |
| speaker_embed_dropout: 0.05 |
| label_smoothing: 0.0 |
| val_check_interval: 5000 |
| check_val_every_n_epoch: 1 |
| precision: "fp16" |
| nworkers: 16 |
| distributed: true |
| accelerator: "ddp" |
| version: null |
| accumulate_grad_batches: 1 |
| use_repetition_token: true |
| use_repetition_gating: false |
| repetition_penalty: 1.0 |
| sampling_temperature: 1.0 |
| top_k: -1 |
| min_top_k: 3 |
| top_p: 0.8 |
| sample_num: 4 |
| length_penalty_max_length: 15000 |
| length_penalty_max_prob: 0.95 |
| max_input_length: 2048 |
| max_output_length: 2000 |
| sample_rate: 16000 |
| n_codes: 1024 |
| n_cluster_groups: 1 |
| phone_context_window: 4 |
| phoneset_size: 1000 |
| inference: |
| top_k: 5 |
|
|
