from local

This commit is contained in:
dohe0342 2023-04-27 16:38:37 +09:00
parent 3192baabdf
commit 3c90779ba8
3 changed files with 3 additions and 1 deletions

View File

@ -82,8 +82,9 @@ class Transducer(nn.Module):
self.sid = sid
if prompt:
#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)))
'''
print(self.prompt)
new_emb = torch.empty(512, 50)
@ -95,6 +96,7 @@ class Transducer(nn.Module):
new_emb = new_emb.transpose(1,0)
self.prompt = torch.nn.Parameter(new_emb)
print(self.prompt)
'''
def forward(
self,