mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-04-26 14:54:37 +09:00
parent 892cce0a1b
commit ee42fa7cb5
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp
View File

Binary file not shown.

8
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py
View File

@ -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,