Avoid deprecation warning for F.tanh (#5413)

b45e65ef · Sam Shleifer · GitHub · 23231c0f · b45e65ef
Unverified Commit b45e65ef authored Jun 30, 2020 by Sam Shleifer Committed by GitHub Jun 30, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/modeling_mobilebert.py src/transformers/modeling_mobilebert.py +1 -1

No files found.
--- 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