jay/pretrain_work_dir/orig_model/configuration.json

{
  "framework": "PyTorch",
  "task" : "text-to-speech",
  "model" : {
    "type" : "sambert-hifigan",
    "lang_type" : "zhcn",
    "sample_rate" : 16000,
    "custom_ckpt": {
      "voice_name" : "F7",
      "am_ckpt" : "basemodel_16k/sambert/ckpt",
      "am_config" : "basemodel_16k/sambert/config.yaml",
      "voc_ckpt" : "basemodel_16k/hifigan/ckpt",
      "voc_config" : "basemodel_16k/hifigan/config.yaml",
      "audio_config" : "basemodel_16k/audio_config_se_16k.yaml",
      "se_model" : "basemodel_16k/speaker_embedding/se.onnx"
    },
    "am": {
       "am": {
          "max_len": 800,

          "embedding_dim": 512, 
          "encoder_num_layers": 8,
          "encoder_num_heads": 8,
          "encoder_num_units": 128,
          "encoder_ffn_inner_dim": 1024,
          "encoder_dropout": 0.1,
          "encoder_attention_dropout": 0.1,
          "encoder_relu_dropout": 0.1,
          "encoder_projection_units": 32,

          "speaker_units": 512,
          "emotion_units": 32,

          "predictor_filter_size": 41,
          "predictor_fsmn_num_layers": 3,
          "predictor_num_memory_units": 128,
          "predictor_ffn_inner_dim": 256,
          "predictor_dropout": 0.1,
          "predictor_shift": 0,
          "predictor_lstm_units": 128,
          "dur_pred_prenet_units": [128, 128],
          "dur_pred_lstm_units": 128,

          "decoder_prenet_units": [256, 256],
          "decoder_num_layers": 12,
          "decoder_num_heads": 8,
          "decoder_num_units": 128,
          "decoder_ffn_inner_dim": 1024,
          "decoder_dropout": 0.1,
          "decoder_attention_dropout": 0.1,
          "decoder_relu_dropout": 0.1,

          "outputs_per_step": 3,
          "num_mels": 82,

          "postnet_filter_size": 41,
          "postnet_fsmn_num_layers": 4,
          "postnet_num_memory_units": 256,
          "postnet_ffn_inner_dim": 512,
          "postnet_dropout": 0.1,
          "postnet_shift": 17,
          "postnet_lstm_units": 128,


          "nsf_f0_global_maximum": 730.0,
          "nsf_f0_global_minimum": 30.0,
          "nsf_norm_type": "global"
      },

      "audio": {
          "frame_shift_ms": 12.5
      },

      "linguistic_unit": {
        "cleaners": "english_cleaners",
        "lfeat_type_list": "sy,tone,syllable_flag,word_segment,emo_category,speaker_category",
        "sy": "dict/sy_dict.txt",
        "tone": "dict/tone_dict.txt",
        "syllable_flag": "dict/syllable_flag_dict.txt",
        "word_segment": "dict/word_segment_dict.txt",
        "emo_category": "dict/emo_category_dict.txt",
        "speaker_category": "dict/speaker_dict.txt"
      },

      "num_gpus": 1,
      "batch_size": 32,
      "group_size": 1024,
      "learning_rate": 0.001,
      "adam_b1": 0.9,
      "adam_b2": 0.98,
      "seed": 1234,

      "num_workers": 4,

      "dist_config": {
          "dist_backend": "nccl",
          "dist_url": "tcp://localhost:11111",
          "world_size": 1
      }

    },
    "vocoder" : {
      "resblock": "1",
      "num_gpus": 1,
      "batch_size": 16,
      "learning_rate": 0.0002,
      "adam_b1": 0.8,
      "adam_b2": 0.99,
      "lr_decay": 0.999,
      "seed": 1234,

      "bias": true,
      "causal": false,
      "nsf_params" : {
        "nb_harmonics": 7, 
        "nsf_f0_global_maximum": 730.0,
        "nsf_f0_global_minimum": 30.0,
        "nsf_norm_type": "global", 
        "sampling_rate": 16000
      },

      "upsample_rates": [10,5,2,2],
      "upsample_kernel_sizes": [20,11,4,4],
      "upsample_initial_channel": 256,
      "resblock_kernel_sizes": [3,7,11],
      "resblock_dilation_sizes": [[1,3,5,7], [1,3,5,7], [1,3,5,7]],

      "segment_size": 6400,
      "num_mels": 80,
      "num_freq": 1025,
      "n_fft": 2048,
      "hop_size": 200,
      "win_size": 1000,

      "sampling_rate": 16000,

      "fmin": 0,
      "fmax": 8000,
      "fmax_for_loss": null,

      "num_workers": 4,

      "dist_config": {
          "dist_backend": "nccl",
          "dist_url": "tcp://localhost:54312",
          "world_size": 1
      }
    }
  },
  "train": {
  },
  "pipeline": {
     "type": "sambert-hifigan-tts"
  }
}