Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Initial support for ContextNet Encoder and CTC Decoder #630

Merged
merged 19 commits into from
May 16, 2020
Merged

Add Initial support for ContextNet Encoder and CTC Decoder #630

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
19 commits
Select commit Hold shift + click to select a range
5f17859
Add SE + context SE support
titu1994 May 10, 2020
aa28939
Add contextnet components
titu1994 May 11, 2020
86e3dbd
Add ContextNet support
titu1994 May 12, 2020
8a5c6de
Add config files
titu1994 May 12, 2020
39c1a91
Correct configs
titu1994 May 12, 2020
28a4cb1
Add streaming speech command
titu1994 May 12, 2020
646dd8f
Add kernel size factor argument
titu1994 May 12, 2020
d990c5c
Add docstrings
titu1994 May 13, 2020
e14d5d5
Update CHANGELOG.md
titu1994 May 13, 2020
d526488
Add integration tests
titu1994 May 13, 2020
851350e
Style fixes and add docstrings for se_reduction_ratio
titu1994 May 13, 2020
2e208f6
Style fixes in tests
titu1994 May 13, 2020
46cc5c7
Correct CHANGELOG.md
titu1994 May 13, 2020
a8d7f4c
Correctios to docstrings
titu1994 May 13, 2020
6d3e4ca
Add WandB support to contextnet.py
titu1994 May 14, 2020
8c81303
Style fixes
titu1994 May 14, 2020
66924b6
Remove unused import
titu1994 May 14, 2020
81330ba
Refactor ContextNetEncoder to subclass JasperEncoder
titu1994 May 14, 2020
7ea9183
Remove unused imports
titu1994 May 14, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Add config files 
Signed-off-by: smajumdar <[email protected]>
  • Loading branch information
@titu1994
titu1994 committed May 13, 2020
commit 8a5c6deb7e169d422aefce9c6357190514a712d2
277 changes: 277 additions & 0 deletions examples/asr/configs/contextnet_128.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,277 @@
model: "ContextNet"
sample_rate: 16000
repeat: &repeat 5
dropout: &dropout 0.0
stride: &stride 1


AudioToTextDataLayer:
max_duration: 16.7
trim_silence: true

train:
shuffle: true

eval:
shuffle: false
max_duration: null

AudioToMelSpectrogramPreprocessor:
window_size: 0.025
window_stride: 0.01
window: "hann"
normalize: "per_feature"
n_fft: 512
features: 80
dither: 0.00001
pad_to: 16
stft_conv: true

SpectrogramAugmentation:
freq_masks: 2
time_masks: 10
freq_width: 27
time_width: 0.05

ContextNetEncoder:
activation: "relu"
conv_mask: false

jasper:
- filters: 128
repeat: 1
kernel: [5]
stride: [1]
dilation: [1]
dropout: 0.0
residual: false
separable: true
se: true

- filters: 128
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 128
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 128
repeat: *repeat
kernel: [5]
stride: [*stride]
dilation: [1]
dropout: *dropout
residual: false
separable: true
se: true
stride_last: true
residual_mode: "stride_add"

- filters: 128
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 128
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 128
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 128
repeat: *repeat
kernel: [5]
stride: [*stride]
dilation: [1]
dropout: *dropout
residual: false
separable: true
se: true
stride_last: true
residual_mode: "stride_add"

- filters: 128
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 128
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 128
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 256
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 256
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 256
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 256
repeat: *repeat
kernel: [5]
stride: [*stride]
dilation: [1]
dropout: *dropout
residual: false
separable: true
se: true
stride_last: true
residual_mode: "stride_add"

- filters: 256
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 256
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 256
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 256
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 256
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 256
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 256
repeat: *repeat
kernel: [5]
stride: [1]
dilation: [1]
dropout: *dropout
residual: true
separable: true
se: true

- filters: 320
repeat: 1
kernel: [5]
stride: [1]
dilation: [1]
dropout: 0.0
residual: false
separable: true

labels: [" ", "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m",
"n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z", "'"]
Loading