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 = [p.reshape(-1, self.input_dim) for p in x]
|
||||||
x = torch.cat(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