From ef3f177e8492c037c0dd438e5fb05ae9d524b562 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 10 Jan 2023 01:21:33 +0900
Subject: [PATCH] from local

---
 .../ASR/incremental_transf/.train.py.swp      | Bin 61440 -> 61440 bytes
 .../ASR/incremental_transf/train.py           |   4 ++--
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/egs/librispeech/ASR/incremental_transf/.train.py.swp b/egs/librispeech/ASR/incremental_transf/.train.py.swp
index e1c275cef6dce5cd435289fb4ce6544e62cd3485..1cb0efa39e4e94ac53174f34441eb68448e4c083 100644
GIT binary patch
delta 231
zcmZw8KMMhI0LJm>{O<fMQnE<ttX5^YYzC7T;F4}q{@lRq1~Kaf7IjQ)a!NMdfh=b{
zlwDo`&kHbo>RV5b<qOLfTiH>{65!bp=QDhTfdlQRN=bTzfIIDJK@%ENoywG;Q;;7T
zQH?TmQGhw=B<L*zN3y6$F^bYv05j4A(kL<V6sJ!BxRK{hmletVu5=|;$aR`#qun!_
W{Z_B5sfIaF?fUxscW|?tXXOW$Of#PV

delta 218
zcmXxXKMMhI0LJm>`rYA7CXv+bHj2#ZQeMOPQ?6SKlELN%7Mt!39MdVG?ge0A@dA_#
zCkwMhx14A3)Tf@W9SA!RJK0q%35Xok{fxr|aG^6bsY*#w=^h78w51JA$f63pV!(~o
zG@=@P72rTqDo~alGT@OxB`Q)VfIXQ4X|#D7RHht(B7eTnC1>_dWuS3gv#0e3W@|X;
R{M)}S4aKQ@M?b%M>OYL%F24W(

diff --git a/egs/librispeech/ASR/incremental_transf/train.py b/egs/librispeech/ASR/incremental_transf/train.py
index bf80fc7ea..08d2dc5a6 100755
--- a/egs/librispeech/ASR/incremental_transf/train.py
+++ b/egs/librispeech/ASR/incremental_transf/train.py
@@ -967,11 +967,11 @@ def run(rank, world_size, args):
     model = get_transducer_model(params)
     
     path1 = '/home/work/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt'
-    pre_trained1 = torch.load(path1)
+    pre_trained1 = torch.load(path1, map_location="cpu")
     pre_trained_model1 = pre_trained1["model"]
     
     path2 = '/home/work/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_24layers.pt'
-    pre_trained2 = torch.load(path2)
+    pre_trained2 = torch.load(path2, map_location="cpu")
     pre_trained_model2 = pre_trained2["model"]
 
     #for n, p in pre_trained_model2.items():