diff --git a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp
index 8a910777d..bf4316098 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 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():