from local

This commit is contained in:
dohe0342 2023-04-11 16:24:38 +09:00
parent 042248251e
commit 46f60bc988
2 changed files with 1 additions and 1 deletions

View File

@ -77,7 +77,7 @@ class Transducer(nn.Module):
nn.LogSoftmax(dim=-1),
)
self.prompt = torch.randn((100, 512), requires_grad=True)
self.prompt = torch.randn((50, 512), requires_grad=True)
def forward(
self,