SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Avoid deprecation warning for F.tanh (#5413)

Browse Source
This commit is contained in:
Sam Shleifer
2020-06-30 16:41:43 -04:00
committed by GitHub
parent 23231c0f78
commit b45e65efa0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_mobilebert.py
View File

@@ -575,7 +575,7 @@ class MobileBertPooler(nn.Module):
return first_token_tensor return first_token_tensor
else: else:
pooled_output = self.dense(first_token_tensor) pooled_output = self.dense(first_token_tensor)
pooled_output = F.tanh(pooled_output) pooled_output = torch.tanh(pooled_output)
return pooled_output return pooled_output