Avoid deprecation warning for F.tanh (#5413)

2020-06-30 16:41:43 -04:00
parent 23231c0f78
commit b45e65efa0
1 changed files with 1 additions and 1 deletions
--- a/src/transformers/modeling_mobilebert.py
+++ b/src/transformers/modeling_mobilebert.py
@@ -575,7 +575,7 @@ class MobileBertPooler(nn.Module):
            return first_token_tensor
        else:
            pooled_output = self.dense(first_token_tensor)
-            pooled_output = F.tanh(pooled_output)
+            pooled_output = torch.tanh(pooled_output)
            return pooled_output