Google Git
Sign in
android / platform / external / pytorch / 29f903aaf2
commit29f903aaf207ea0a108bbdf1373189574a4986bd[log] [tgz]
authorPriya Goyal <prigoyal@fb.com>Fri Dec 16 07:51:04 2016 -0800
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>Fri Dec 16 07:59:25 2016 -0800
tree77fb752b007f0883115c45d3b015c514e9a71321
parente8dc09064eab84124ce606ef82c30f5dbd59e50a [diff]
Make computed params broadcast optional

Summary: this was introduced due to rm and riv params in SpatialBN layer and the likes. We should be saving these params as well but it is not required to broadcast these params to all gpus after every epoch.

Differential Revision: D4338749

fbshipit-source-id: d3bbc92cf0cd7d220a51d76aea8bffcfd6e520b7
1 file changed
tree: 77fb752b007f0883115c45d3b015c514e9a71321
  1. caffe/
  2. caffe2/
  3. docs/
  4. third_party/
  5. .Doxyfile
  6. .gitignore
  7. .gitmodules
  8. build.py
  9. build_android.py
  10. build_android_prepare.py
  11. LICENSE
  12. Makefile
  13. README.md
README.md

Caffe2

Caffe2 is a deep learning framework made with expression, speed, and modularity in mind. It is an experimental refactoring of Caffe, and allows a more flexible way to organize computation.

Read the installation instructions for installation details.

License and Citation

Caffe2 is released under the BSD 2-Clause license.