from local

2022-12-23 17:13:50 +09:00 · 2022-12-23 17:13:50 +09:00 · 5178627904
commit 5178627904
parent 525e1c0982
2 changed files with 2 additions and 0 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless5/.train.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless5/.train.py.swp
--- a/egs/librispeech/ASR/pruned_transducer_stateless5/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless5/train.py
@ -965,6 +965,8 @@ def run(rank, world_size, args):

    logging.info("About to create model")
    model = get_transducer_model(params)
+    logging.info(model)
+    exit()

    num_param = sum([p.numel() for p in model.parameters()])
    logging.info(f"Number of model parameters: {num_param}")