facebookarchive/caffe2
MKL-DNN integration does not support backward pass
オープン
#2,376 opened on 2018/03/22
help wanted
Repository metrics
- Stars
- (8,424 個のスター)
- PR merge metrics
- (PR metrics pending)
説明
MKL-DNN integration (in caffe2/mkl) implements only forward pass operators, although MKL-DNN itself provides primitives for both forward and backward pass. Implementation of ConvGradient operator would let Caffe2 use MKL-DNN for training on CPU, and support some models which need backprop (e.g. deep dream).