diff --git a/egs/librispeech/ASR/pruned_transducer_stateless6/.train.py.swo b/egs/librispeech/ASR/pruned_transducer_stateless6/.train.py.swo index f7c95cbe2..d7c090d3a 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless6/.train.py.swo and b/egs/librispeech/ASR/pruned_transducer_stateless6/.train.py.swo differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless6/train.py b/egs/librispeech/ASR/pruned_transducer_stateless6/train.py index 71a70cb88..9600a8c3c 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless6/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless6/train.py @@ -929,6 +929,7 @@ def run(rank, world_size, args): logging.info("About to create model") model = get_transducer_model(params) + logging.info(model) num_param = sum([p.numel() for p in model.parameters()]) logging.info(f"Number of model parameters: {num_param}")