From d942cecf2000bfe656c20182fed05a8ab00094e2 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Mon, 9 Jan 2023 22:48:42 +0900
Subject: [PATCH] from local

---
 .../incremental_transf/.identity_train.py.swp | Bin 77824 -> 77824 bytes
 .../ASR/incremental_transf/identity_train.py  |   7 ++++---
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp
index 8a910777d619c5259c16f988e0a377394a38e932..bf431609885ce6b87127dc73dff09119f8d91bc0 100644
GIT binary patch
delta 129
zcmZp8z|!!5MKsAE%+puFQqO<^2m}}yWTp2cpWY}sTaeLl^G3m7J>GgY28Q|U5J|_$
zf)2k$tvDDM*ns!|P!uFH4Tz_1R&*?3<5vKIlA^@CVh}@la=yC)qw?f_cg4x#9x6bV
RwTJoSmmZdzt36tu0|3jBAmIQ2

delta 125
zcmZp8z|!!5MKsAE%+puFQqO<^2m}}yM5Xs6FWo3QTaeLV^G3m7J>G3>3=EgpA(9T0
z1s#5insYEPd;*GqMBf1MJs{q&S<$hCZF09K*W|12f|DCPMJCs~3rueI6avy>P;t}A
N+#cqei#+<C0{}B~B@F-o

diff --git a/egs/librispeech/ASR/incremental_transf/identity_train.py b/egs/librispeech/ASR/incremental_transf/identity_train.py
index a4df22fae..4765edfb4 100755
--- a/egs/librispeech/ASR/incremental_transf/identity_train.py
+++ b/egs/librispeech/ASR/incremental_transf/identity_train.py
@@ -980,9 +980,10 @@ def run(rank, world_size, args):
         load_checkpoint(path, transducer_model)
     except:
         path = '/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)
-    transducer_model.to(device)
+        load_checkpoint(path, transducer_model)
+    #pre_trained_model = pre_trained_model['model']
+    #transducer_model.load_state_dict(pre_trained_model, strict=True)
+    #transducer_model.to(device)
     model = get_interformer_model(transducer_model.encoder, params)
 
     for n, p in model.named_parameters():