csrc/amp_C_frontend.cpp · 7a428776bb94a231af6111855a52afbb1b4d604c · OpenDAS / apex

luise.chen authored Mar 24, 2023

* Add fused_lars optimizer

* Update primitive fused_lars optimizer, working for resnet50 with NHWC/NCHW

* Add flow of using nesterov in FusedLARS

7a428776

amp_C_frontend.cpp 5.77 KB