This is an automated email from the ASF dual-hosted git repository.
haibin pushed a change to branch benchmark
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from a47b540 multi-precision lamb update operator
add cc0c356 squad multi-lamb
No new revisions were added by this update.
Summary of changes:
python/mxnet/ndarray/contrib.py | 24 +++
python/mxnet/optimizer/optimizer.py | 92 ++++++++-
python/mxnet/test_utils.py | 80 +++++---
src/operator/contrib/multi_lamb-inl.h | 332 ++++++++++++++++++++++++++++++++
src/operator/contrib/multi_lamb.cc | 245 +++++++++++++++++++++++
src/operator/contrib/multi_lamb.cu | 254 ++++++++++++++++++++++++
tests/python/unittest/test_optimizer.py | 98 ++++++++++
7 files changed, 1096 insertions(+), 29 deletions(-)
create mode 100644 src/operator/contrib/multi_lamb-inl.h
create mode 100644 src/operator/contrib/multi_lamb.cc
create mode 100644 src/operator/contrib/multi_lamb.cu
