tests/test_modeling_t5.py · d533c7e9b9a5122cb1381702b99a3aff0aa2e0c4 · chenpangpang / transformers · GitLab

Find file Blame History Permalink

[fix] T5 ONNX test: model.to(torch_device) (#5769) · d533c7e9
Funtowicz Morgan authored Jul 15, 2020
```
Signed-off-by: Morgan Funtowicz <morgan@huggingface.co>
```
d533c7e9

test_modeling_t5.py 38.3 KB