Minor fixes.

This commit is contained in:
Fangjun Kuang 2021-12-30 00:15:23 +08:00
parent 8541dc73f9
commit 7828c6ff73

View File

@ -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