This is an automated email from the ASF dual-hosted git repository. anirudh2290 pushed a change to branch master in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 72c180c Correct ONNX documentation (#15914) add acc074f Add AMP Conversion support for BucketingModule (#15528) No new revisions were added by this update. Summary of changes: docs/tutorials/amp/amp_tutorial.md | 2 +- example/rnn/bucketing/README.md | 6 + example/rnn/bucketing/cudnn_rnn_bucketing.py | 19 ++- python/mxnet/contrib/amp/amp.py | 64 ++++++++++ python/mxnet/model.py | 33 +++--- python/mxnet/module/bucketing_module.py | 170 +++++++++++++++++++++++++-- python/mxnet/module/module.py | 2 +- tests/python/gpu/test_contrib_amp.py | 35 ++++++ tests/python/train/test_bucketing.py | 50 +++++--- tests/python/unittest/test_module.py | 71 +++++++++++ 10 files changed, 405 insertions(+), 47 deletions(-)