diff --git a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp
index 882671d85..15f5829fe 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 831bf3f62..874dbdbbd 100755
--- a/egs/librispeech/ASR/incremental_transf/identity_train.py
+++ b/egs/librispeech/ASR/incremental_transf/identity_train.py
@@ -975,8 +975,11 @@ def run(rank, world_size, args):
     logging.info("About to create model")
     transducer_model = get_transducer_model(params)
     
-    try: pre_trained_model = torch.load('/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt')
-    except: pre_trained_model = torch.load('/home/work/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt')
+    try: 
+        path = '/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt'
+        load_checkpoint(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)
@@ -987,7 +990,6 @@ def run(rank, world_size, args):
             p.requires_grad = False
         else:
             print(n)
-    exit()
     '''
     for n, p in model.named_parameters():
         if 'layer' not in n: