from local

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

View File

@ -90,7 +90,7 @@ class Transducer(nn.Module):
new_emb[i] = torch.normal(float(mean), float(std), size=50).squeeze() new_emb[i] = torch.normal(float(mean), float(std), size=50).squeeze()
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)
def forward( def forward(
self, self,