saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Tanh torch warnings

Browse source
This commit is contained in:
Aryansh Omray 2020-04-16 20:53:46 +05:30 committed by Julien Chaumond
parent 16469fedbd
commit 14cdeee75a
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/transformers/activations.py
View file

@ -47,7 +47,7 @@ ACT2FN = {
"relu": F.relu,
"swish": swish,
"gelu": gelu,
"tanh": F.tanh,
"tanh": torch.tanh,
"gelu_new": gelu_new,
}