54 lines
1.0 KiB
YAML
54 lines
1.0 KiB
YAML
# This is an example that demonstrates how to configure a model file.
|
|
# You can modify the configuration according to your own requirements.
|
|
|
|
# to print the register_table:
|
|
# from funasr.register import tables
|
|
# tables.print()
|
|
|
|
model: CTTransformer
|
|
model_conf:
|
|
ignore_id: 0
|
|
embed_unit: 256
|
|
att_unit: 256
|
|
dropout_rate: 0.1
|
|
punc_list:
|
|
- <unk>
|
|
- _
|
|
- ','
|
|
- 。
|
|
- '?'
|
|
- 、
|
|
punc_weight:
|
|
- 1.0
|
|
- 1.0
|
|
- 1.0
|
|
- 1.0
|
|
- 1.0
|
|
- 1.0
|
|
sentence_end_id: 3
|
|
|
|
encoder: SANMEncoder
|
|
encoder_conf:
|
|
input_size: 256
|
|
output_size: 256
|
|
attention_heads: 8
|
|
linear_units: 1024
|
|
num_blocks: 4
|
|
dropout_rate: 0.1
|
|
positional_dropout_rate: 0.1
|
|
attention_dropout_rate: 0.0
|
|
input_layer: pe
|
|
pos_enc_class: SinusoidalPositionEncoder
|
|
normalize_before: true
|
|
kernel_size: 11
|
|
sanm_shfit: 0
|
|
selfattention_layer_type: sanm
|
|
padding_idx: 0
|
|
|
|
tokenizer: CharTokenizer
|
|
tokenizer_conf:
|
|
unk_symbol: <unk>
|
|
|
|
|
|
|