icefall_asr_tedlium3_pruned_transducer_stateless / log /greedy_search /log-decode-epoch-29-avg-13-context-2-max-sym-per-frame-3-2022-03-21-16-33-45
luomingshuang's picture
add tedlium3-pruned-transducer-stateless files
70628de
raw
history blame
4.54 kB
2022-03-21 16:33:45,405 INFO [decode.py:427] Decoding started
2022-03-21 16:33:45,405 INFO [decode.py:433] Device: cuda:0
2022-03-21 16:33:45,411 INFO [decode.py:443] {'feature_dim': 80, 'subsampling_factor': 4, 'attention_dim': 512, 'nhead': 8, 'dim_feedforward': 2048, 'num_encoder_layers': 12, 'vgg_frontend': False, 'embedding_dim': 512, 'env_info': {'k2-version': '1.13', 'k2-build-type': 'Release', 'k2-with-cuda': True, 'k2-git-sha1': '5ee082ea55f50e8bd42203ba266945ea5a236ab8', 'k2-git-date': 'Sun Feb 27 09:00:48 2022', 'lhotse-version': '1.0.0.dev+git.d917411.clean', 'torch-cuda-available': True, 'torch-cuda-version': '10.1', 'python-version': '3.8', 'icefall-git-branch': 'tedlium3-pruned-transducer-stateless-recipe', 'icefall-git-sha1': 'ad28c8c-dirty', 'icefall-git-date': 'Fri Mar 18 11:39:06 2022', 'icefall-path': '/ceph-meixu/luomingshuang/icefall', 'k2-path': '/ceph-meixu/luomingshuang/k2/k2/python/k2/__init__.py', 'lhotse-path': '/ceph-meixu/luomingshuang/anaconda3/envs/k2-python/lib/python3.8/site-packages/lhotse-1.0.0.dev0+git.d917411.clean-py3.8.egg/lhotse/__init__.py', 'hostname': 'de-74279-k2-train-2-0307200233-b554c565c-lf9qd', 'IP address': '10.177.74.201'}, 'epoch': 29, 'avg': 13, 'exp_dir': PosixPath('pruned_transducer_stateless/exp'), 'bpe_model': 'data/lang_bpe_500/bpe.model', 'decoding_method': 'greedy_search', 'beam_size': 4, 'context_size': 2, 'max_sym_per_frame': 3, 'manifest_dir': PosixPath('data/fbank'), 'max_duration': 200.0, 'bucketing_sampler': True, 'num_buckets': 30, 'concatenate_cuts': False, 'duration_factor': 1.0, 'gap': 1.0, 'on_the_fly_feats': False, 'shuffle': True, 'return_cuts': True, 'num_workers': 2, 'enable_spec_aug': True, 'spec_aug_time_warp_factor': 80, 'enable_musan': True, 'res_dir': PosixPath('pruned_transducer_stateless/exp/greedy_search'), 'suffix': 'epoch-29-avg-13-context-2-max-sym-per-frame-3', 'blank_id': 0, 'unk_id': 2, 'vocab_size': 500}
2022-03-21 16:33:45,411 INFO [decode.py:445] About to create model
2022-03-21 16:33:46,108 INFO [decode.py:456] averaging ['pruned_transducer_stateless/exp/epoch-17.pt', 'pruned_transducer_stateless/exp/epoch-18.pt', 'pruned_transducer_stateless/exp/epoch-19.pt', 'pruned_transducer_stateless/exp/epoch-20.pt', 'pruned_transducer_stateless/exp/epoch-21.pt', 'pruned_transducer_stateless/exp/epoch-22.pt', 'pruned_transducer_stateless/exp/epoch-23.pt', 'pruned_transducer_stateless/exp/epoch-24.pt', 'pruned_transducer_stateless/exp/epoch-25.pt', 'pruned_transducer_stateless/exp/epoch-26.pt', 'pruned_transducer_stateless/exp/epoch-27.pt', 'pruned_transducer_stateless/exp/epoch-28.pt', 'pruned_transducer_stateless/exp/epoch-29.pt']
2022-03-21 16:34:07,582 INFO [decode.py:465] Number of model parameters: 84514780
2022-03-21 16:34:07,582 INFO [asr_datamodule.py:357] About to get dev cuts
2022-03-21 16:34:07,617 INFO [asr_datamodule.py:362] About to get test cuts
2022-03-21 16:34:07,686 INFO [asr_datamodule.py:300] About to create dev dataset
2022-03-21 16:34:07,688 INFO [asr_datamodule.py:319] About to create dev dataloader
2022-03-21 16:34:11,811 INFO [decode.py:352] batch 0/?, cuts processed until now is 22
2022-03-21 16:35:58,689 INFO [decode.py:369] The transcripts are stored in pruned_transducer_stateless/exp/greedy_search/recogs-dev-greedy_search-epoch-29-avg-13-context-2-max-sym-per-frame-3.txt
2022-03-21 16:35:58,722 INFO [utils.py:406] [dev-greedy_search] %WER 7.27% [1325 / 18226, 171 ins, 491 del, 663 sub ]
2022-03-21 16:35:58,806 INFO [decode.py:382] Wrote detailed error stats to pruned_transducer_stateless/exp/greedy_search/errs-dev-greedy_search-epoch-29-avg-13-context-2-max-sym-per-frame-3.txt
2022-03-21 16:35:58,807 INFO [decode.py:399]
For dev, WER of different settings are:
greedy_search 7.27 best for dev
2022-03-21 16:36:02,945 INFO [decode.py:352] batch 0/?, cuts processed until now is 29
2022-03-21 16:38:54,255 INFO [decode.py:369] The transcripts are stored in pruned_transducer_stateless/exp/greedy_search/recogs-test-greedy_search-epoch-29-avg-13-context-2-max-sym-per-frame-3.txt
2022-03-21 16:38:54,300 INFO [utils.py:406] [test-greedy_search] %WER 6.69% [1902 / 28430, 197 ins, 802 del, 903 sub ]
2022-03-21 16:38:54,399 INFO [decode.py:382] Wrote detailed error stats to pruned_transducer_stateless/exp/greedy_search/errs-test-greedy_search-epoch-29-avg-13-context-2-max-sym-per-frame-3.txt
2022-03-21 16:38:54,400 INFO [decode.py:399]
For test, WER of different settings are:
greedy_search 6.69 best for test
2022-03-21 16:38:54,400 INFO [decode.py:491] Done!