From b45e65efa0fbff2611ddd68e14fa75cacef3fe08 Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Tue, 30 Jun 2020 16:41:43 -0400 Subject: [PATCH] Avoid deprecation warning for F.tanh (#5413) --- src/transformers/modeling_mobilebert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_mobilebert.py b/src/transformers/modeling_mobilebert.py index 4cbaf3a502..9b29514836 100644 --- 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