| dataset: |
| bpe_model: checkpoints/bpe.model |
| sample_rate: 24000 |
| squeeze: false |
| mel: |
| sample_rate: 24000 |
| n_fft: 1024 |
| hop_length: 256 |
| win_length: 1024 |
| n_mels: 100 |
| mel_fmin: 0 |
| normalize: false |
|
|
| gpt: |
| model_dim: 1024 |
| max_mel_tokens: 605 |
| max_text_tokens: 402 |
| heads: 16 |
| use_mel_codes_as_input: true |
| mel_length_compression: 1024 |
| layers: 20 |
| number_text_tokens: 12000 |
| number_mel_codes: 8194 |
| start_mel_token: 8192 |
| stop_mel_token: 8193 |
| start_text_token: 0 |
| stop_text_token: 1 |
| train_solo_embeddings: false |
| condition_type: "conformer_perceiver" |
| condition_module: |
| output_size: 512 |
| linear_units: 2048 |
| attention_heads: 8 |
| num_blocks: 6 |
| input_layer: "conv2d2" |
| perceiver_mult: 2 |
|
|
| vqvae: |
| channels: 100 |
| num_tokens: 8192 |
| hidden_dim: 512 |
| num_resnet_blocks: 3 |
| codebook_dim: 512 |
| num_layers: 2 |
| positional_dims: 1 |
| kernel_size: 3 |
| smooth_l1_loss: true |
| use_transposed_convs: false |
|
|
| bigvgan: |
| adam_b1: 0.8 |
| adam_b2: 0.99 |
| lr_decay: 0.999998 |
| seed: 1234 |
|
|
| resblock: "1" |
| upsample_rates: [4,4,4,4,2,2] |
| upsample_kernel_sizes: [8,8,4,4,4,4] |
| upsample_initial_channel: 1536 |
| resblock_kernel_sizes: [3,7,11] |
| resblock_dilation_sizes: [[1,3,5], [1,3,5], [1,3,5]] |
| feat_upsample: false |
| speaker_embedding_dim: 512 |
| cond_d_vector_in_each_upsampling_layer: true |
|
|
| gpt_dim: 1024 |
|
|
| activation: "snakebeta" |
| snake_logscale: true |
|
|
| use_cqtd_instead_of_mrd: true |
| cqtd_filters: 128 |
| cqtd_max_filters: 1024 |
| cqtd_filters_scale: 1 |
| cqtd_dilations: [1, 2, 4] |
| cqtd_hop_lengths: [512, 256, 256] |
| cqtd_n_octaves: [9, 9, 9] |
| cqtd_bins_per_octaves: [24, 36, 48] |
|
|
| resolutions: [[1024, 120, 600], [2048, 240, 1200], [512, 50, 240]] |
| mpd_reshapes: [2, 3, 5, 7, 11] |
| use_spectral_norm: false |
| discriminator_channel_mult: 1 |
|
|
| use_multiscale_melloss: true |
| lambda_melloss: 15 |
|
|
| clip_grad_norm: 1000 |
|
|
| segment_size: 16384 |
| num_mels: 100 |
| num_freq: 1025 |
| n_fft: 1024 |
| hop_size: 256 |
| win_size: 1024 |
|
|
| sampling_rate: 24000 |
|
|
| fmin: 0 |
| fmax: null |
| fmax_for_loss: null |
| mel_type: "pytorch" |
|
|
| num_workers: 2 |
| dist_config: |
| dist_backend: "nccl" |
| dist_url: "tcp://localhost:54321" |
| world_size: 1 |
|
|
| dvae_checkpoint: dvae.pth |
| gpt_checkpoint: gpt.pth |
| bigvgan_checkpoint: bigvgan_generator.pth |
|
|
