[WIP]Add AMP + Use Boolean Mask + Update Benchmarking Script

dmlc / gluon-nlp

NLP made easy

https://nlp.gluon.ai/

Apache License 2.0

2.55k stars 538 forks source link

Closed sxjscience closed 3 years ago

sxjscience commented 3 years ago

Add AMP

[x] PR's title starts with a category (e.g. [BUGFIX], [MODEL], [TUTORIAL], [FEATURE], [DOC], etc)
[ ] Changes are complete (i.e. I finished coding on this PR)
[ ] All changes have test coverage
[ ] Code is well-documented

cc @dmlc/gluon-nlp-team