mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
Minor fixes.
This commit is contained in:
parent
8541dc73f9
commit
7828c6ff73
@ -65,8 +65,8 @@ class Joiner(nn.Module):
|
||||
x = [p.reshape(-1, self.input_dim) for p in x]
|
||||
x = torch.cat(x)
|
||||
|
||||
logit = torch.tanh(x)
|
||||
activations = torch.tanh(x)
|
||||
|
||||
output = self.output_linear(logit)
|
||||
logits = self.output_linear(activations)
|
||||
|
||||
return output
|
||||
return logits
|
||||
|
Loading…
x
Reference in New Issue
Block a user