diff --git a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp index 3f36fbd7b..f9a81e2ee 100644 Binary files a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp and b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp differ diff --git a/egs/librispeech/ASR/incremental_transf/identity_train.py b/egs/librispeech/ASR/incremental_transf/identity_train.py index 4e4d18091..b8b768d7a 100755 --- a/egs/librispeech/ASR/incremental_transf/identity_train.py +++ b/egs/librispeech/ASR/incremental_transf/identity_train.py @@ -975,12 +975,14 @@ def run(rank, world_size, args): logging.info(params) logging.info("About to create model") - model = get_transducer_model(params) + transducer_model = get_transducer_model(params) pre_trained_model = torch.load('/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt') pre_trained_model = pre_trained_model['model'] - model.load_state_dict(pre_trained_model, strict=True) - print(model) + transducer_model.load_state_dict(pre_trained_model, strict=True) + + model = get_interformer_model(transducer_model.encoder, params) + print(model) ''' for n, p in model.named_parameters(): if 'layer' not in n: