commit | a5f44ed2653d4b8d90ea84b285951bad92b5ea68 | [log] [tgz] |
---|---|---|
author | Pooya Davoodi <pdavoodi@nvidia.com> | Thu May 25 13:13:11 2017 -0700 |
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Thu May 25 13:18:23 2017 -0700 |
tree | 8a7e01c00522681d521e7966ecbd966e68c205a9 | |
parent | e35a4fe5cc17d8754fe1e7172f68fdb901c23651 [diff] |
Fix number of indices and block_size in SparseAdam Summary: Fix number of indices and block_size in SparseAdam to support gradients of any dimension. Closes https://github.com/caffe2/caffe2/pull/249 Reviewed By: asaadaldien Differential Revision: D5125714 Pulled By: akyrola fbshipit-source-id: 84134049cb9a77e58562272ea351222befe27fca
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.
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.
Detailed build matrix (hit refresh if you see icons not showing up due to heroku):
Target | Status |
---|---|
Linux | |
Mac (CPU) | |
Android | |
iOS | |
Linux + MKL | |
Windows |