tests/test_modeling_funnel.py · c65863ce5341c3dac162fb5e301ba7f6bee800fa · chenpangpang / transformers · GitLab

Find file Blame History Permalink

Fix FP16 and attention masks in FunnelTransformer (#7374) · ad39271a
Sylvain Gugger authored Sep 25, 2020
```
* Fix #7371

* Fix training

* Fix test values

* Apply the fix to TF as well
```
ad39271a

test_modeling_funnel.py 16.4 KB