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-27 20:33:57 +09:00
parent c39ee49207
commit 9609b959dc
3 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

Binary file not shown.

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

Binary file not shown.

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

@ -84,10 +84,9 @@ class Transducer(nn.Module):
print(self.sid) print(self.sid)
print('-'*20) print('-'*20)
if prompt: if prompt:
#self.prompt = torch.randn((200, 512), requires_grad=True) self.prompt = torch.randn((200, 512), requires_grad=True)
#statistic = open(f'/home/work/workspace/icefall/egs/librispeech/ASR/conv_feat/{self.sid}/{sid}_statistic.txt', 'r').readlines() statistic = open(f'/home/work/workspace/icefall/egs/librispeech/ASR/conv_feat/{self.sid}/{sid}_statistic.txt', 'r').readlines()
self.prompt = torch.nn.Parameter(torch.rand((50, 512))) self.prompt = torch.nn.Parameter(torch.rand((50, 512)))
'''
print(self.prompt) print(self.prompt)
new_emb = torch.empty(512, 50) new_emb = torch.empty(512, 50)
@ -99,7 +98,6 @@ class Transducer(nn.Module):
new_emb = new_emb.transpose(1,0) new_emb = new_emb.transpose(1,0)
self.prompt = torch.nn.Parameter(new_emb) self.prompt = torch.nn.Parameter(new_emb)
print(self.prompt) print(self.prompt)
'''
def forward( def forward(
self, self,