| commit | c05c500a82edb48673dfc6250b765ef523a32e05 | [log] [tgz] |
|---|---|---|
| author | Aapo Kyrola <akyrola@fb.com> | Mon Aug 14 19:46:33 2017 -0700 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Mon Aug 14 19:47:59 2017 -0700 |
| tree | c26bcd4924f1b644d2f442ad45eb86622d1641eb | |
| parent | 62dcc2feed4251135659f363df3d02d11ad58348 [diff] |
check _grad suffix Summary: Memonger had a subtle bug which caused it to recycle "splitinfo" outputs of Concat/Split. That is bad since they are in CPU device, and woult cause them to be realloaced. This caused big slowdown with Kaiming's trainer. Bug was that we checked for gradients as contaning "_grad" in the name, although we should only allow it as a suffix. Admittedly, this is not elegant to do string checking anyways, but that is how Caffe2 works now. Reviewed By: asaadaldien Differential Revision: D5627251 fbshipit-source-id: c12be2323109bf81c3725d8884c7ef024e010bd5
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.