icefall-asr-spgispeech-pruned-transducer-stateless2 / log /fast_beam_search /log-decode-iter-696000-avg-10-beam-4-max-contexts-4-max-states-8-2022-05-14-17-37-44
desh2608's picture
updated logs with CER
3179a80
2022-05-14 17:37:44,304 INFO [decode.py:503] Decoding started
2022-05-14 17:37:44,304 INFO [decode.py:509] Device: cuda:0
2022-05-14 17:37:44,360 INFO [decode.py:518] {'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, 'encoder_dim': 512, 'nhead': 8, 'dim_feedforward': 2048, 'num_encoder_layers': 12, 'decoder_dim': 512, 'joiner_dim': 512, 'model_warm_step': 3000, 'env_info': {'k2-version': '1.14', 'k2-build-type': 'Debug', 'k2-with-cuda': True, 'k2-git-sha1': '1b29f0a946f50186aaa82df46a59f492ade9692b', 'k2-git-date': 'Tue Apr 12 20:46:49 2022', 'lhotse-version': '1.1.0', 'torch-version': '1.10.1+cu111', 'torch-cuda-available': True, 'torch-cuda-version': '11.1', 'python-version': '3.8', 'icefall-git-branch': 'spgi', 'icefall-git-sha1': '4fc1638-dirty', 'icefall-git-date': 'Sat May 14 10:41:06 2022', 'icefall-path': '/exp/draj/mini_scale_2022/icefall', 'k2-path': '/exp/draj/mini_scale_2022/k2/k2/python/k2/__init__.py', 'lhotse-path': '/exp/draj/mini_scale_2022/lhotse/lhotse/__init__.py', 'hostname': 'r8n04', 'IP address': '10.1.8.4'}, 'epoch': 20, 'iter': 696000, 'avg': 10, 'exp_dir': PosixPath('pruned_transducer_stateless2/exp/v2'), 'bpe_model': 'data/lang_bpe_500/bpe.model', 'decoding_method': 'fast_beam_search', 'beam_size': 4, 'beam': 4, 'max_contexts': 4, 'max_states': 8, 'context_size': 2, 'max_sym_per_frame': 1, 'manifest_dir': PosixPath('data/manifests'), 'enable_musan': True, 'concatenate_cuts': False, 'duration_factor': 1.0, 'gap': 1.0, 'max_duration': 500, 'num_buckets': 30, 'on_the_fly_feats': False, 'shuffle': True, 'num_workers': 8, 'enable_spec_aug': True, 'spec_aug_time_warp_factor': 80, 'res_dir': PosixPath('pruned_transducer_stateless2/exp/v2/fast_beam_search'), 'suffix': 'iter-696000-avg-10-beam-4-max-contexts-4-max-states-8', 'blank_id': 0, 'vocab_size': 500}
2022-05-14 17:37:44,360 INFO [decode.py:520] About to create model
2022-05-14 17:37:45,039 INFO [decode.py:537] averaging ['pruned_transducer_stateless2/exp/v2/checkpoint-696000.pt', 'pruned_transducer_stateless2/exp/v2/checkpoint-688000.pt', 'pruned_transducer_stateless2/exp/v2/checkpoint-680000.pt', 'pruned_transducer_stateless2/exp/v2/checkpoint-672000.pt', 'pruned_transducer_stateless2/exp/v2/checkpoint-664000.pt', 'pruned_transducer_stateless2/exp/v2/checkpoint-656000.pt', 'pruned_transducer_stateless2/exp/v2/checkpoint-648000.pt', 'pruned_transducer_stateless2/exp/v2/checkpoint-640000.pt', 'pruned_transducer_stateless2/exp/v2/checkpoint-632000.pt', 'pruned_transducer_stateless2/exp/v2/checkpoint-624000.pt']
2022-05-14 17:39:25,683 INFO [decode.py:562] Number of model parameters: 78648040
2022-05-14 17:39:25,683 INFO [asr_datamodule.py:337] About to get SPGISpeech dev cuts
2022-05-14 17:39:25,692 INFO [asr_datamodule.py:342] About to get SPGISpeech val cuts