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(-)
