kce/model.yaml

audio_net_config:
  input_trans:
    act: ReLU
    dim:
    - 256
    - 256
    norm: LayerNorm
    num_block: 1
  transformer_config:
    conv_config:
      channels: 256
    corss_att_config:
      n_feats: 256
      n_head: 1
      norm: LayerNorm
    cross_att: MultiHeadCrossAtt
    feed_forward_config: &id001
      act: ReLU
      dim:
      - 256
      - 2048
      - 256
      norm: LayerNorm
      num_block: 1
    self_att: MultiHeadAtt
    self_att_config: &id002
      n_feats: 256
      n_head: 4
    size: 256
decoder_net_config:
  input_trans:
    act: ReLU
    dim: 256
    norm: LayerNorm
    num_block: 1
  transformer_config:
    corss_att_config:
      n_feats: 256
      n_head: 1
      norm: LayerNorm
    cross_att: MultiHeadCrossAtt
    feed_forward_config: *id001
    self_att: MultiHeadAtt
    self_att_config: *id002
    size: 256
eos: 5004
kw_net_config:
  input_trans:
    act: ReLU
    dim: 256
    norm: LayerNorm
    num_block: 1
  num_bpe_token: 5005
  num_phn_token: 74
  transformer_config:
    feed_forward_config: *id001
    self_att: MultiHeadAtt
    self_att_config: *id002
    size: 256
loss_weight:
- 0.3
- 0.1
- 0.3
num_audio_block: 9
num_doecoder_block: 4
num_kw_block: 4
sos: 2
sv_net_config:
  front_output_size: 256
  num_classes: 2338
  pooling_conf:
    use_softmax: False
    layer_indices:
    - 0
    - 1
    - 2
    - 3
    - 4
    - 5
    - 6
    - 7
    - 8
    type: learnable_weights