from local

This commit is contained in:
dohe0342 2023-04-26 14:55:04 +09:00
parent ee42fa7cb5
commit 27e90ab9eb
2 changed files with 2 additions and 0 deletions

View File

@ -88,6 +88,8 @@ class Transducer(nn.Module):
for i in range(512):
mean, std = statistic[i].strip().split(' ')
new_emb[i] = torch.normal(float(mean), float(std), size=50).squeeze()
new_emb = new_emb.transpose(1,0)
self.prompt = torch.nn.Parameter(new_emb)
def forward(