Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • luise.chen's avatar
    Add FusedLARS optimizer (#109) · e519c1e3
    luise.chen authored 
    * Add fused_lars optimizer

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

* Add flow of using nesterov in FusedLARS
    e519c1e3
multi_tensor_lars.cu 9.8 KB