mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 10:02:22 +00:00
Code Issues Packages Projects Releases Wiki Activity
465 Commits 5 Branches 3 Tags
Commit Graph

10 Commits

Author SHA1 Message Date
Daniel Povey
4e23fb2252
Improve diagnostics code memory-wise and accumulate more stats. (#373) 
* Update diagnostics, hopefully print more stats.

# Conflicts:
#	egs/librispeech/ASR/pruned_transducer_stateless4b/train.py

* Remove memory-limit options arg

* Remove unnecessary option for diagnostics code, collect on more batches
 2022-05-19 11:45:59 +08:00
Fangjun Kuang
fce7f3cd9a
Support computing RNN-T loss with torchaudio (#316) 2022-04-19 18:47:13 +08:00
Fangjun Kuang
9a11808ed3
Set the seed for dataloader. (#282) 
Also, suppress torch warnings about division by truncation.
 2022-03-31 16:48:46 +08:00
yaozengwei
ad62981765
Add diagnostics (#230) 
* Adding diagnostics code...

* Move diagnostics code from local dir to the shared icefall dir

* Remove the diagnostics code in the local dir

* Update docs of arguments, and remove stats_types() function in TensorDiagnosticOptions object.

* Update docs of arguments.

* Add copyright information.

* Corrected the time in copyright information.

Co-authored-by: Daniel Povey <dpovey@gmail.com>
 2022-03-04 15:38:23 +08:00
Fangjun Kuang
1c35ae1dba
Reset seed at the beginning of each epoch. (#221) 
* Reset seed at the beginning of each epoch.

* Use a different seed for each epoch.
 2022-02-21 15:16:39 +08:00
Fangjun Kuang
a8150021e0
Use modified transducer loss in training. (#179) 
* Use modified transducer loss in training.

* Minor fix.

* Add modified beam search.

* Add modified beam search.

* Minor fixes.

* Fix typo.

* Update RESULTS.

* Fix a typo.

* Minor fixes.
 2022-02-07 18:37:36 +08:00
Wei Kang
5ae80dfca7
Minor fixes (#193) 2022-01-27 18:01:17 +08:00
Fangjun Kuang
14c93add50
Remove batchnorm, weight decay, and SOS from transducer conformer encoder (#155) 
* Remove batchnorm, weight decay, and SOS.

* Make --context-size configurable.

* Update results.
 2021-12-27 16:01:10 +08:00
Fangjun Kuang
fb6a57e9e0
Increase the size of the context in the RNN-T decoder. (#153) 2021-12-23 07:55:02 +08:00
Fangjun Kuang
1d44da845b
RNN-T Conformer training for LibriSpeech (#143) 
* Begin to add RNN-T training for librispeech.

* Copy files from conformer_ctc.

Will edit it.

* Use conformer/transformer model as encoder.

* Begin to add training script.

* Add training code.

* Remove long utterances to avoid OOM when a large max_duraiton is used.

* Begin to add decoding script.

* Add decoding script.

* Minor fixes.

* Add beam search.

* Use LSTM layers for the encoder.

Need more tunings.

* Use stateless decoder.

* Minor fixes to make it ready for merge.

* Fix README.

* Update RESULT.md to include RNN-T Conformer.

* Minor fixes.

* Fix tests.

* Minor fixes.

* Minor fixes.

* Fix tests.
 2021-12-18 07:42:51 +08:00