deepspeech2.yaml 1.2 KB
Newer Older
H
Hui Zhang 已提交
1 2 3 4 5
# https://yaml.org/type/float.html
data:
  train_manifest: data/manifest.train
  dev_manifest: data/manifest.dev
  test_manifest: data/manifest.test
H
Hui Zhang 已提交
6
  mean_std_filepath: data/mean_std.json
H
Hui Zhang 已提交
7
  vocab_filepath: data/vocab.txt 
H
Hui Zhang 已提交
8
  augmentation_config: conf/augmentation.json
H
Hui Zhang 已提交
9
  batch_size: 64 # one gpu
H
Hui Zhang 已提交
10 11 12 13 14 15
  min_input_len: 0.0
  max_input_len: 27.0 # second
  min_output_len: 0.0
  max_output_len: 400.0
  min_output_input_ratio: 0.05
  max_output_input_ratio: 10.0
H
Hui Zhang 已提交
16 17 18 19 20 21
  specgram_type: linear
  target_sample_rate: 16000
  max_freq: None
  n_fft: None
  stride_ms: 10.0
  window_ms: 20.0
H
Hui Zhang 已提交
22 23
  delta_delta: False
  dither: 1.0
H
Hui Zhang 已提交
24 25 26 27 28 29 30
  use_dB_normalization: True
  target_dB: -20
  random_seed: 0
  keep_transcription_text: False
  sortagrad: True
  shuffle_method: batch_shuffle
  num_workers: 0
H
Hui Zhang 已提交
31

H
Hui Zhang 已提交
32 33 34 35 36 37
model:
  num_conv_layers: 2
  num_rnn_layers: 3
  rnn_layer_size: 1024
  use_gru: True 
  share_rnn_weights: False
H
Hui Zhang 已提交
38

H
Hui Zhang 已提交
39
training:
H
Hui Zhang 已提交
40 41
  n_epoch: 50
  lr: 2e-3
H
Hui Zhang 已提交
42 43 44
  lr_decay: 0.83
  weight_decay: 1e-06
  global_grad_clip: 5.0
H
Hui Zhang 已提交
45
  log_interval: 100
H
Hui Zhang 已提交
46
  
H
Hui Zhang 已提交
47 48 49 50 51
decoding:
  batch_size: 128
  error_rate_type: cer 
  decoding_method: ctc_beam_search
  lang_model_path: data/lm/zh_giga.no_cna_cmn.prune01244.klm
H
Hui Zhang 已提交
52
  alpha: 1.9
H
Hui Zhang 已提交
53 54 55 56 57
  beta: 5.0
  beam_size: 300
  cutoff_prob: 0.99
  cutoff_top_n: 40
  num_proc_bsearch: 10