commit | 6ed36c37e63c7b9821c0bc993448abb8204d0771 | [log] [tgz] |
---|---|---|
author | Aapo Kyrola <akyrola@fb.com> | Thu Apr 20 15:02:09 2017 -0700 |
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Thu Apr 20 15:02:48 2017 -0700 |
tree | 0c5854dfb0e5afe3eec606ab4ad3b15759f755b4 | |
parent | 4f2531fbaa865cddf79b05b0611d32770e70818c [diff] |
fix CUDNN layer weight size calculation for multiple layers Summary: CuDNN LSTM weights were incorrectly sized for layers > 0: there was assumption that the input size to middle layers is same as for the first layer, but actually the middle layer will get input from a layer below, which will have dimension equal to the output dimension (hidden dimension). This worked fine when input_dim and hidden_dim were equal, as are the default params for lstm_benchmark. Reviewed By: salexspb Differential Revision: D4922824 fbshipit-source-id: 3ed05529dcb0a4e66ad440084a55df1c5932fd33
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.
Target | Status |
---|---|
Linux | |
Android | |
iOS | |
Linux + MKL |