Use tanh in the joint network.

This commit is contained in:
Fangjun Kuang 2021-12-18 11:14:05 +08:00
parent 66cc9b4592
commit 63e1266e3a

View File

@ -48,7 +48,7 @@ class Joiner(nn.Module):
# Now decoder_out is (N, 1, U, C)
logit = encoder_out + decoder_out
logit = F.relu(logit)
logit = F.tanh(logit)
output = self.output_linear(logit)