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 19ce5b161..d57f63b17 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp differ 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 6eccafaf3..b3813d3a6 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py @@ -80,7 +80,15 @@ class Transducer(nn.Module): self.prompt = None if prompt: #self.prompt = torch.randn((200, 512), requires_grad=True) + statistic = open('/home/work/workspace/icefall/egs/librispeech/ASR/conv_feat/2094_statistic.txt', 'r').readlines() self.prompt = torch.nn.Parameter(torch.rand((50, 512))) + print(self.prompt) + + new_emb = torch.empty(512, 50) + for i in range(512): + mean, std = statistic[i].strip().split(' ') + new_emb[i] = torch.normal(float(mean), float(std), size=50).squeeze() + def forward( self,