csrc/amp_C_frontend.cpp · e519c1e3ccc98d709b7a37f3c32f4003f1a00b50 · 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

e519c1e3

amp_C_frontend.cpp 5.77 KB