mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 01:52:41 +00:00
Add commands for reproducing. (#40)
* Add commands for reproducing. * Use --bucketing-sampler by default.
This commit is contained in:
parent
abadc71415
commit
7f8e3a673a
@ -21,6 +21,32 @@ To get more unique paths, we scaled the lattice.scores with 0.5 (see https://git
|
|||||||
|test-clean|1.3|1.2|
|
|test-clean|1.3|1.2|
|
||||||
|test-other|1.2|1.1|
|
|test-other|1.2|1.1|
|
||||||
|
|
||||||
|
You can use the following commands to reproduce our results:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
git clone https://github.com/k2-fsa/icefall
|
||||||
|
cd icefall
|
||||||
|
|
||||||
|
# It was using ef233486, you may not need to switch to it
|
||||||
|
# git checkout ef233486
|
||||||
|
|
||||||
|
cd egs/librispeech/ASR
|
||||||
|
./prepare.sh
|
||||||
|
|
||||||
|
export CUDA_VISIBLE_DEVICES="0,1,2,3"
|
||||||
|
python conformer_ctc/train.py --bucketing-sampler True \
|
||||||
|
--concatenate-cuts False \
|
||||||
|
--max-duration 200 \
|
||||||
|
--full-libri True \
|
||||||
|
--world-size 4
|
||||||
|
|
||||||
|
python conformer_ctc/decode.py --lattice-score-scale 0.5 \
|
||||||
|
--epoch 34 \
|
||||||
|
--avg 20 \
|
||||||
|
--method attention-decoder \
|
||||||
|
--max-duration 20 \
|
||||||
|
--num-paths 100
|
||||||
|
```
|
||||||
|
|
||||||
### LibriSpeech training results (Tdnn-Lstm)
|
### LibriSpeech training results (Tdnn-Lstm)
|
||||||
#### 2021-08-24
|
#### 2021-08-24
|
||||||
|
@ -82,14 +82,14 @@ class LibriSpeechAsrDataModule(DataModule):
|
|||||||
group.add_argument(
|
group.add_argument(
|
||||||
"--max-duration",
|
"--max-duration",
|
||||||
type=int,
|
type=int,
|
||||||
default=500.0,
|
default=200.0,
|
||||||
help="Maximum pooled recordings duration (seconds) in a "
|
help="Maximum pooled recordings duration (seconds) in a "
|
||||||
"single batch. You can reduce it if it causes CUDA OOM.",
|
"single batch. You can reduce it if it causes CUDA OOM.",
|
||||||
)
|
)
|
||||||
group.add_argument(
|
group.add_argument(
|
||||||
"--bucketing-sampler",
|
"--bucketing-sampler",
|
||||||
type=str2bool,
|
type=str2bool,
|
||||||
default=False,
|
default=True,
|
||||||
help="When enabled, the batches will come from buckets of "
|
help="When enabled, the batches will come from buckets of "
|
||||||
"similar duration (saves padding frames).",
|
"similar duration (saves padding frames).",
|
||||||
)
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user