| commit | 60c17a34b5422cdaea8314aca1f7956cc0350f62 | [log] [tgz] |
|---|---|---|
| author | Aapo Kyrola <akyrola@fb.com> | Tue Aug 29 18:56:26 2017 -0700 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Tue Aug 29 19:11:08 2017 -0700 |
| tree | 558ad432ee57f0d58fcf2a24bc51bb406f477b27 | |
| parent | 41adebe974da45e1ed47dc6b8aae12386cf97b53 [diff] |
better default settings for CUB Summary: The default CUB settings led to very slow execution in practice when using "dynamic" memory allocation with C2 (i.e freeing blobs after their use). After some tinkering, I arrived to these numbers, that work with resnet-50 and NVIDIA M40 GPU much better than the origianal defaults. Also made the maximum allocated memory configurable. Reviewed By: Yangqing Differential Revision: D5732930 fbshipit-source-id: 9ff34f49d5a3eb138bc6f44c82918731a35325a6
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.