| commit | fefd5479a3ad02cd30a223d6d7ae8b1d6af0fd18 | [log] [tgz] |
|---|---|---|
| author | Wojciech Glogowski <glogowski@fb.com> | Wed Aug 30 18:46:17 2017 -0700 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Wed Aug 30 18:53:46 2017 -0700 |
| tree | af703747b859ddf274eaca8278d5725ade902a8b | |
| parent | 5ed5be71b109345f8025fa6590d81151b9039062 [diff] |
Initial implementation of YellowFin algorithm Summary: Added YellowFin optimizer to Caffe2. This implemention is different from the original: It has separate alpha and mu for each parameter and it uses different version of Momentum SGD. Tests / benchmarks for the optimizer are to be done. Some refactor of the code is to be done before pushing. This is still a working version. Reviewed By: akyrola Differential Revision: D5652689 fbshipit-source-id: c10dc0424f47c3051b454aede1d121902cb759a8
Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.
Caffe2 research award competition request for proposals
Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.
Please participate in our survey (https://www.surveymonkey.com/r/caffe2). We will send you information about new releases and special developer events/webinars.
Caffe2 is released under the BSD 2-Clause license.