Closed sxjscience closed 3 years ago
Switch to use boolean mask, which helps reduce the memory consumption of the models.
cc @dmlc/gluon-nlp-team
The documentation website for preview: http://gluon-nlp-staging.s3-accelerate.dualstack.amazonaws.com/PR1402/boolean_mask/index.html
Description
Switch to use boolean mask, which helps reduce the memory consumption of the models.
Checklist
Essentials
cc @dmlc/gluon-nlp-team