Tanh torch warnings

2020-04-16 20:53:46 +05:30
parent 16469fedbd
commit 14cdeee75a
1 changed files with 1 additions and 1 deletions
--- a/src/transformers/activations.py
+++ b/src/transformers/activations.py
@@ -47,7 +47,7 @@ ACT2FN = {
    "relu": F.relu,
    "swish": swish,
    "gelu": gelu,
-    "tanh": F.tanh,
+    "tanh": torch.tanh,
    "gelu_new": gelu_new,
 }