src/transformers/trainer.py · 0106826a65cfecb7c21d4dc525d72799868e0380 · chenpangpang / transformers · GitLab

Find file Blame History Permalink

Fix missing autocast() in Trainer.prediction_step() (#14075) · 0106826a
Kwanghee Choi authored Oct 20, 2021
```
Co-authored-by: jonas <jonas@hpcnt.com>
```
0106826a

trainer.py 131 KB