Skip to content

Commit 00c48ec

Browse files
authored
Model average (#344)
* First upload of model average codes. * minor fix * update decode file * update .flake8 * rename pruned_transducer_stateless3 to pruned_transducer_stateless4 * change epoch number counter starting from 1 instead of 0 * minor fix of pruned_transducer_stateless4/train.py * refactor the checkpoint.py * minor fix, update docs, and modify the epoch number to count from 1 in the pruned_transducer_stateless4/decode.py * update author info * add docs of the scaling in function average_checkpoints_with_averaged_model
1 parent 8635fb4 commit 00c48ec

File tree

15 files changed

+1811
-1
lines changed

15 files changed

+1811
-1
lines changed

.flake8

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ per-file-ignores =
99
egs/tedlium3/ASR/*/conformer.py: E501,
1010
egs/gigaspeech/ASR/*/conformer.py: E501,
1111
egs/librispeech/ASR/pruned_transducer_stateless2/*.py: E501,
12+
egs/librispeech/ASR/pruned_transducer_stateless4/*.py: E501,
1213
egs/librispeech/ASR/*/optim.py: E501,
1314
egs/librispeech/ASR/*/scaling.py: E501,
1415

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
../pruned_transducer_stateless2/__init__.py
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
../pruned_transducer_stateless2/asr_datamodule.py
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
../pruned_transducer_stateless2/beam_search.py
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
../pruned_transducer_stateless2/conformer.py

0 commit comments

Comments
 (0)