template.yaml 1.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. # This is an example that demonstrates how to configure a model file.
  2. # You can modify the configuration according to your own requirements.
  3. # to print the register_table:
  4. # from funasr.register import tables
  5. # tables.print()
  6. model: CTTransformerStreaming
  7. model_conf:
  8. ignore_id: 0
  9. embed_unit: 256
  10. att_unit: 256
  11. dropout_rate: 0.1
  12. punc_list:
  13. - <unk>
  14. - _
  15. - ,
  16. - 。
  17. - ?
  18. - 、
  19. punc_weight:
  20. - 1.0
  21. - 1.0
  22. - 1.0
  23. - 1.0
  24. - 1.0
  25. - 1.0
  26. sentence_end_id: 3
  27. encoder: SANMVadEncoder
  28. encoder_conf:
  29. input_size: 256
  30. output_size: 256
  31. attention_heads: 8
  32. linear_units: 1024
  33. num_blocks: 3
  34. dropout_rate: 0.1
  35. positional_dropout_rate: 0.1
  36. attention_dropout_rate: 0.0
  37. input_layer: pe
  38. pos_enc_class: SinusoidalPositionEncoder
  39. normalize_before: true
  40. kernel_size: 11
  41. sanm_shfit: 5
  42. selfattention_layer_type: sanm
  43. padding_idx: 0
  44. tokenizer: CharTokenizer
  45. tokenizer_conf:
  46. unk_symbol: <unk>