tests/test_activations.py · 877fc56410e3a0495f62e07e66a73e6b3b9629bc · chenpangpang / transformers · GitLab

Find file Blame History Permalink

get_activation('relu') provides a simple mapping from strings i… (#2807) · ef74b0f0
Sam Shleifer authored Feb 13, 2020
```
* activations.py contains a mapping from string to activation function
* resolves some `gelu` vs `gelu_new` ambiguity
```
ef74b0f0

test_activations.py 863 Bytes