From 8c7c3a0171ff2f82fa0821f9c9af5a42e737cff9 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 11 Apr 2023 16:25:31 +0900 Subject: [PATCH] from local --- .../.model.py.swp | Bin 28672 -> 28672 bytes .../.prompt_tuning.py.swp | Bin 94208 -> 94208 bytes .../model.py | 5 +++-- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp index d43c3d510782698aec6b927cad8b1483bdb0e3ce..a74a78c0d8a35b4f5231d21c456ca11c4f852789 100644 GIT binary patch delta 159 zcmZp8z}WDBQ7p+I%+puFQqO<^2m}}yvbarC7OmVU_KTk}db6OwXMWz-ObiUgED%|s zkb*s<;d)jEhB-hS4#Z(V>71>zJS4g+FaAeIMWW*~kFRDKMImjf|KYxm|x#ZOF=+x?X$m#Ms* RoUCd&*(pPO^Ep)`F96e#9~J-r diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp index 2b3fb91a844d8c8a717aba83895212f5965cf7ca..85f1188a3968c8c5261b13fcd8816e105ff2eb93 100644 GIT binary patch delta 35 pcmZp8z}oPDRW!*U%+puFQqO<^2m}}y61YuMavMdriZZ^=2LP>E2~hw5 delta 35 pcmZp8z}oPDRW!*U%+puFQqO<^2m}}y?72--92!NpiZZ^=2LP&N2?GEC diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py index 35799ba0a..a5e63c088 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py @@ -76,8 +76,9 @@ class Transducer(nn.Module): nn.Linear(encoder_dim, vocab_size), nn.LogSoftmax(dim=-1), ) - - self.prompt = torch.randn((50, 512), requires_grad=True) + + if prompt: + self.prompt = torch.randn((50, 512), requires_grad=True) def forward( self,