Li Peng
|
3e4da5f781
|
Replace deprecated pytorch methods (#1814)
* Replace deprecated pytorch methods
- torch.cuda.amp.GradScaler(...) => torch.amp.GradScaler("cuda", ...)
- torch.cuda.amp.autocast(...) => torch.amp.autocast("cuda", ...)
* Replace `with autocast(...)` with `with autocast("cuda", ...)`
Co-authored-by: Li Peng <lipeng@unisound.ai>
|
2024-12-16 10:24:16 +08:00 |
|
Fangjun Kuang
|
7b0afbdc16
|
Remove cur_batch_idx (#1102)
|
2023-05-30 14:49:54 +08:00 |
|
Desh Raj
|
d31db01037
|
manual correction of black formatting
|
2022-11-17 14:18:05 -05:00 |
|
Desh Raj
|
107df3b115
|
apply black on all files
|
2022-11-17 09:42:17 -05:00 |
|
Fangjun Kuang
|
60317120ca
|
Revert "Apply new Black style changes"
|
2022-11-17 20:19:32 +08:00 |
|
Desh Raj
|
d110b04ad3
|
apply new black formatting to all files
|
2022-11-16 13:06:43 -05:00 |
|
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 |
|
Desh Raj
|
5aafbb970e
|
SPGISpeech recipe (#334)
* initial commit for SPGISpeech recipe
* add decoding
* add spgispeech transducer
* remove conformer ctc; minor fixes in RNN-T
* add results
* add tensorboard
* add pretrained model to HF
* remove unused scripts and soft link common scripts
* remove duplicate files
* pre commit hooks
* remove change in librispeech
* pre commit hook
* add CER numbers
|
2022-05-16 20:52:14 +08:00 |
|