model.json

{
  "model" : {
    "fm_decoder_downsampling_factor" : "1,2,4,2,1",
    "fm_decoder_num_layers" : "2,2,4,4,4",
    "fm_decoder_cnn_module_kernel" : "31,15,7,15,31",
    "fm_decoder_feedforward_dim" : 1536,
    "fm_decoder_num_heads" : 4,
    "fm_decoder_dim" : 512,
    "text_encoder_downsampling_factor" : "1",
    "text_encoder_num_layers" : "4",
    "text_encoder_feedforward_dim" : 512,
    "text_encoder_cnn_module_kernel" : 9,
    "text_encoder_num_heads" : 4,
    "text_encoder_dim" : 192,
    "query_head_dim" : 32,
    "value_head_dim" : 32,
    "pos_head_dim" : 4,
    "pos_dim" : 48,
    "time_embed_dim" : 192,
    "text_embed_dim" : 192
  },
  "feature" : {
    "sampling_rate": 24000,
    "feat_dim": 100,
    "n_fft" : 1024,
    "hop_length" : 256
  }
}