diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py index 23619b258..5019a7f9f 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -1009,9 +1009,6 @@ def run(rank, world_size, args): num_param = sum([p.numel() for p in model.parameters()]) logging.info(f"Number of model parameters: {num_param}") - for name, module in model.named_modules(): - num_param = sum([p.numel() for p in module.parameters()]) - logging.info(f"Number of model parameters for {name}: {num_param}") assert params.save_every_n >= params.average_period model_avg: Optional[nn.Module] = None