train_ctc_sate_conformer.yaml 1.08 KB
Newer Older
1 2
train-subset: train_st
valid-subset: dev_st
xuchen committed
3 4 5 6 7 8 9 10 11 12 13 14

max-epoch: 50
max-update: 100000

num-workers: 8
patience: 10
no-progress-bar: True
log-interval: 100
seed: 1
report-accuracy: True

#load-pretrained-encoder-from:
15 16
#load-pretrained-acoustic-encoder-from:
#load-pretrained-text-encoder-from:
xuchen committed
17 18
#load-pretrained-decoder-from:

19
arch: s2t_sate
xuchen committed
20 21 22 23 24 25 26 27 28 29 30 31 32
share-decoder-input-output-embed: True
optimizer: adam
clip-norm: 10.0
lr-scheduler: inverse_sqrt
warmup-init-lr: 1e-7
warmup-updates: 10000
lr: 2e-3
#adam_betas: (0.9,0.98)

ctc-weight: 0.3
criterion: label_smoothed_cross_entropy_with_ctc
label_smoothing: 0.1

33 34
encoder-normalize-before: True
decoder-normalize-before: True
xuchen committed
35 36 37 38 39 40 41
conv-kernel-sizes: 5,5
conv-channels: 1024
dropout: 0.1
activation-fn: relu
encoder-embed-dim: 256
encoder-ffn-embed-dim: 2048
encoder-layers: 12
42
text-encoder-layers: 6
xuchen committed
43 44 45 46 47 48 49
decoder-layers: 6
encoder-attention-heads: 4

macaron-style: True
use-cnn-module: True
cnn-module-kernel: 31

50 51 52
acoustic-encoder: conformer
adapter: league

xuchen committed
53 54 55 56 57
#decoder-embed-dim: 256
#decoder-ffn-embed-dim: 2048
#decoder-attention-heads: 4
#attention-dropout: 0.1
#activation-dropout: 0.1