From a6a7a9e63decc84837371a3eb6985f809947e025 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Mon, 9 Jan 2023 22:38:29 +0900 Subject: [PATCH] from local --- .../incremental_transf/.identity_train.py.swp | Bin 77824 -> 77824 bytes .../ASR/incremental_transf/identity_train.py | 4 +++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp index 1b670576be0ce7bcdd07d0bcdce8efa675a2b166..e39de7911f9c09994373e6b2b66e27e30862a0bd 100644 GIT binary patch delta 371 zcmXBQJuCxp7{~GF^iF$PFE^A#mm9<+TtrkmX+kjAM6jIFO2lg}VykvC6^WZ{UJ^V1 zMYNN!+@uUPniz=HXz?w-qJJs(9=@7Z@sxDt%#VeMU zRy8X*pouaw-_3?iL}`Jh)L@aciH@SRbn76u)MzS0 zFOje98iRiTVU;wI+b}R;sCUzM^4gwf_>`JvscG8T>zHW z^izCx%J8*HfA@6f3-eB18!M4$amNzbjCRgb?Kf%-iUc^vF*cA!0^|4{5P5}Oij#xRO7CPfKiBG)*cl=dJp%u=e#JL*;})`hP$L&uJQN}P?17C 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