diff --git a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp index 1b670576b..e39de7911 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 3e9c56035..56f93c676 100755 --- a/egs/librispeech/ASR/incremental_transf/identity_train.py +++ b/egs/librispeech/ASR/incremental_transf/identity_train.py @@ -979,8 +979,10 @@ def run(rank, world_size, args): except: pre_trained_model = torch.load('/home/work/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt') pre_trained_model = pre_trained_model['model'] transducer_model.load_state_dict(pre_trained_model, strict=True) - model = get_interformer_model(transducer_model.encoder, params) + + del pre_trained_model + for n, p in model.named_parameters(): if 'pt_encoder' in n: p.requires_grad = False