File size: 6,130 Bytes
101cbee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
2023-05-03 12:08:37,757 INFO [decode.py:777] Decoding started
2023-05-03 12:08:37,757 INFO [decode.py:783] Device: cuda:0
2023-05-03 12:08:37,760 INFO [decode.py:793] {'frame_shift_ms': 10.0, 'allowed_excess_duration_ratio': 0.1, 'best_train_loss': inf, 'best_valid_loss': inf, 'best_train_epoch': -1, 'best_valid_epoch': -1, 'batch_idx_train': 0, 'log_interval': 50, 'reset_interval': 200, 'valid_interval': 3000, 'feature_dim': 80, 'subsampling_factor': 4, 'warm_step': 2000, 'env_info': {'k2-version': '1.23.4', 'k2-build-type': 'Release', 'k2-with-cuda': True, 'k2-git-sha1': 'a23383c5a381713b51e9014f3f05d096f8aceec3', 'k2-git-date': 'Wed Apr 26 15:33:33 2023', 'lhotse-version': '1.14.0.dev+git.b61b917.dirty', 'torch-version': '1.13.1', 'torch-cuda-available': True, 'torch-cuda-version': '11.6', 'python-version': '3.1', 'icefall-git-branch': 'master', 'icefall-git-sha1': '45c13e9-dirty', 'icefall-git-date': 'Mon Apr 24 15:00:02 2023', 'icefall-path': '/k2-dev/yangyifan/icefall-master', 'k2-path': '/k2-dev/yangyifan/anaconda3/envs/icefall/lib/python3.10/site-packages/k2-1.23.4.dev20230427+cuda11.6.torch1.13.1-py3.10-linux-x86_64.egg/k2/__init__.py', 'lhotse-path': '/k2-dev/yangyifan/anaconda3/envs/icefall/lib/python3.10/site-packages/lhotse-1.14.0.dev0+git.b61b917.dirty-py3.10.egg/lhotse/__init__.py', 'hostname': 'de-74279-k2-train-1-1220091118-57c4d55446-mlpzc', 'IP address': '10.177.22.19'}, 'epoch': 30, 'iter': 0, 'avg': 18, 'use_averaged_model': True, 'exp_dir': PosixPath('pruned_transducer_stateless7/exp_multidataset'), 'bpe_model': 'data/lang_bpe_500/bpe.model', 'lang_dir': PosixPath('data/lang_bpe_500'), 'decoding_method': 'greedy_search', 'beam_size': 4, 'beam': 20.0, 'ngram_lm_scale': 0.01, 'max_contexts': 8, 'max_states': 64, 'context_size': 2, 'max_sym_per_frame': 1, 'num_paths': 200, 'nbest_scale': 0.5, 'use_shallow_fusion': False, 'lm_type': 'rnn', 'lm_scale': 0.3, 'tokens_ngram': 3, 'backoff_id': 500, 'num_encoder_layers': '2,4,3,2,4', 'feedforward_dims': '1024,1024,2048,2048,1024', 'nhead': '8,8,8,8,8', 'encoder_dims': '384,384,384,384,384', 'attention_dims': '192,192,192,192,192', 'encoder_unmasked_dims': '256,256,256,256,256', 'zipformer_downsampling_factors': '1,2,4,8,2', 'cnn_module_kernels': '31,31,31,31,31', 'decoder_dim': 512, 'joiner_dim': 512, 'full_libri': True, 'manifest_dir': PosixPath('data/fbank'), 'cv_manifest_dir': PosixPath('data/en/fbank'), 'max_duration': 600, 'bucketing_sampler': True, 'num_buckets': 30, 'concatenate_cuts': False, 'duration_factor': 1.0, 'gap': 1.0, 'on_the_fly_feats': False, 'shuffle': True, 'drop_last': True, 'return_cuts': True, 'num_workers': 2, 'enable_spec_aug': True, 'spec_aug_time_warp_factor': 80, 'enable_musan': True, 'input_strategy': 'PrecomputedFeatures', 'lm_vocab_size': 500, 'lm_epoch': 7, 'lm_avg': 1, 'lm_exp_dir': None, 'rnn_lm_embedding_dim': 2048, 'rnn_lm_hidden_dim': 2048, 'rnn_lm_num_layers': 3, 'rnn_lm_tie_weights': True, 'transformer_lm_exp_dir': None, 'transformer_lm_dim_feedforward': 2048, 'transformer_lm_encoder_dim': 768, 'transformer_lm_embedding_dim': 768, 'transformer_lm_nhead': 8, 'transformer_lm_num_layers': 16, 'transformer_lm_tie_weights': True, 'res_dir': PosixPath('pruned_transducer_stateless7/exp_multidataset/greedy_search'), 'suffix': 'epoch-30-avg-18-context-2-max-sym-per-frame-1-use-averaged-model', 'blank_id': 0, 'unk_id': 2, 'vocab_size': 500}
2023-05-03 12:08:37,761 INFO [decode.py:795] About to create model
2023-05-03 12:08:38,505 INFO [zipformer.py:178] At encoder stack 4, which has downsampling_factor=2, we will combine the outputs of layers 1 and 3, with downsampling_factors=2 and 8.
2023-05-03 12:08:38,519 INFO [decode.py:862] Calculating the averaged model over epoch range from 12 (excluded) to 30
2023-05-03 12:08:48,268 INFO [decode.py:924] Number of model parameters: 70369391
2023-05-03 12:08:48,269 INFO [asr_datamodule.py:449] About to get test-clean cuts
2023-05-03 12:08:48,272 INFO [asr_datamodule.py:456] About to get test-other cuts
2023-05-03 12:08:52,265 INFO [decode.py:674] batch 0/?, cuts processed until now is 44
2023-05-03 12:09:42,223 INFO [decode.py:688] The transcripts are stored in pruned_transducer_stateless7/exp_multidataset/greedy_search/recogs-test-clean-epoch-30-avg-18-context-2-max-sym-per-frame-1-use-averaged-model.txt
2023-05-03 12:09:42,321 INFO [utils.py:558] [test-clean-greedy_search] %WER 2.02% [1060 / 52576, 93 ins, 126 del, 841 sub ]
2023-05-03 12:09:42,514 INFO [decode.py:699] Wrote detailed error stats to pruned_transducer_stateless7/exp_multidataset/greedy_search/errs-test-clean-epoch-30-avg-18-context-2-max-sym-per-frame-1-use-averaged-model.txt
2023-05-03 12:09:42,514 INFO [decode.py:713]
For test-clean, WER of different settings are:
greedy_search 2.02 best for test-clean
2023-05-03 12:09:44,163 INFO [decode.py:674] batch 0/?, cuts processed until now is 52
2023-05-03 12:10:21,381 INFO [zipformer.py:1454] attn_weights_entropy = tensor([4.1026, 3.1372, 3.7565, 2.0714, 3.1314, 2.3136, 3.4924, 3.3003],
device='cuda:0'), covar=tensor([0.0254, 0.1036, 0.0419, 0.2082, 0.0721, 0.0983, 0.0632, 0.1135],
device='cuda:0'), in_proj_covar=tensor([0.0147, 0.0154, 0.0160, 0.0147, 0.0138, 0.0124, 0.0137, 0.0166],
device='cuda:0'), out_proj_covar=tensor([0.0004, 0.0004, 0.0004, 0.0004, 0.0004, 0.0003, 0.0004, 0.0004],
device='cuda:0')
2023-05-03 12:10:28,241 INFO [decode.py:688] The transcripts are stored in pruned_transducer_stateless7/exp_multidataset/greedy_search/recogs-test-other-epoch-30-avg-18-context-2-max-sym-per-frame-1-use-averaged-model.txt
2023-05-03 12:10:28,344 INFO [utils.py:558] [test-other-greedy_search] %WER 4.30% [2252 / 52343, 214 ins, 221 del, 1817 sub ]
2023-05-03 12:10:28,550 INFO [decode.py:699] Wrote detailed error stats to pruned_transducer_stateless7/exp_multidataset/greedy_search/errs-test-other-epoch-30-avg-18-context-2-max-sym-per-frame-1-use-averaged-model.txt
2023-05-03 12:10:28,552 INFO [decode.py:713]
For test-other, WER of different settings are:
greedy_search 4.3 best for test-other
2023-05-03 12:10:28,552 INFO [decode.py:958] Done!
|