You can use the following command to test the pretrained.py and the pretrained files:
CUDA_VISIBLE_DEVICES='1' python pruned_transducer_stateless/pretrained.py --checkpoint icefall_asr_tedlium3_pruned_transducer_stateless/exp/pretrained_average_17_to_29.pt --bpe-model icefall_asr_tedlium3_pruned_transducer_stateless/data/lang_bpe_500/bpe.model --method greedy_search icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W01.wav icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W02.wav icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W03.wav
The running results are as follows:
2022-03-21 16:19:46,125 INFO [pretrained.py:253] {'sample_rate': 16000, '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'}, 'checkpoint': 'icefall_asr_tedlium3_pruned_transducer_stateless/exp/pretrained_average_17_to_29.pt', 'bpe_model': 'icefall_asr_tedlium3_pruned_transducer_stateless/data/lang_bpe_500/bpe.model', 'method': 'greedy_search', 'sound_files': ['icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W01.wav', 'icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W02.wav', 'icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W03.wav'], 'beam_size': 4, 'context_size': 2, 'max_sym_per_frame': 3, 'blank_id': 0, 'unk_id': 2, 'vocab_size': 500}
2022-03-21 16:19:46,125 INFO [pretrained.py:259] device: cuda:0
2022-03-21 16:19:46,125 INFO [pretrained.py:261] Creating model
2022-03-21 16:20:03,351 INFO [pretrained.py:270] Constructing Fbank computer
2022-03-21 16:20:03,354 INFO [pretrained.py:280] Reading sound files: ['icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W01.wav', 'icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W02.wav', 'icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W03.wav']
2022-03-21 16:20:03,379 INFO [pretrained.py:286] Decoding started
2022-03-21 16:20:03,519 INFO [pretrained.py:306] Using greedy_search
2022-03-21 16:20:05,040 INFO [pretrained.py:334]
icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W01.wav:
choice isn 't it i don 't live there but i did journey on a twenty seven thousand mile trip for two years to the fastest growing and widest counties in america what is a whitopia i define whitopia in three ways first a whitopia has posted at least six percent
icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W02.wav:
population growth since two thousand secondly the majority of that growth comes from white migrants and third a whitopia has an ineffable charm a pleasant look and feel a jenesequa to learn how and why whitopias are ticking i immersed myself for several months apiece in three of them first
icefall_asr_tedlium3_pruned_transducer_stateless/test_wavs/RichBenjamin_2015W03.wav:
st george utah second coeuralene idaho and third for siteth county georgia first stop st george a beautiful town of red rock landscapes in the 1850s brigham young dispatched families to st george to grow cotton because of the hot arid climate and so they called it utah 's dixie and the name sticks to this day
2022-03-21 16:20:05,040 INFO [pretrained.py:336] Decoding Done