| commit | bcc8c8f696879ef2cac9e9bb4cd84ba4cc7d0041 | [log] [tgz] |
|---|---|---|
| author | Matthew Chan <matthewchan@fb.com> | Wed Nov 08 16:32:19 2017 -0800 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Wed Nov 08 16:43:18 2017 -0800 |
| tree | 37b26aec7cdcff1470967b1f7b72c1f97770c683 | |
| parent | 4e3aa25139746305a613ec5730273467746ec519 [diff] |
Support RMSProp in Caffe2. Summary: Add `RmsPropOptimizer` to `optimizer.py` so RMSProp can be used as an optimizer. `RmpsPropOptimizer` uses `RmpPropOp` to update the gradient and `MomentumSGDUpdateOp` to update the model parameters. Differential Revision: D6118279 fbshipit-source-id: e38b8380ff74c1d1bb1e87fc300b6b55e32cd2e0
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 Apache 2.0 license. See the NOTICE file for details.