[FIX] MASTER AMP and onnxruntime issue with master PT by felixdittrich92 路 Pull Request #986 路 mindee/doctr (original) (raw)
This PR:
- fix amp usage for master arch in both frameworks
- fix onnxruntime issue with MASTER PT (instead of tril triu with transpose works fine)
Any feedback is welcome 馃
Closes:
#789
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks! Only a few comments
slow tests passes / amp tested also again
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks Felix :)
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.Learn more about bidirectional Unicode characters
[ Show hidden characters]({{ revealButtonHref }})