Skip to content

Conversation

@i8run
Copy link
Contributor

@i8run i8run commented Oct 11, 2016

fix #23 . Some problems exist in spatial convolution and concat .

psyyz10 and others added 27 commits November 16, 2016 09:54
Because of some unknown reasons, the back propagation method in Concat
is not `updateGradInput`, but `backward` instead, which should not
override in the class inherited from module. So the testcases in concat
should adopt to the situation.
1. Add some testcases for layers which use mkl dnn api. There are some
   testcases in WebScaleML. Alghough it has been passed all of testcases
   of WebScaleML, for some big input, like Convolution from GoogLeNet,
   AlexNet, the result may be wrong. Based on current testcases, we
   found that we must do more test for float and big input.

2. Fix the bug about wrong result of gradInput of Pooling (Max and
   Avg), because MKL-DNN will not erase the data existing in gradInput.

3. Fix the bug about wrong result when some layers in concat layer are
   not MKL-DNN layer.

4. Note, because the different implementation of layers between MKL-DNN
   and Spark-DL, the result is not always same for convolution, lrn and
   batch norm. So the output and gradInput of AlexNet, GoogLeNet v1 and
   GoogLeNet v2 are not completely same with SparkDL w/ MKL-Blas.
   Currently, the error we set may be 1e-4~1e-5.

   We need some convergence test for the implementation of MKL-DNN.
1. memset optmized with openmp
2. omit double conversion
3. fix backward filter and bias of convolution, which will get wrong
   answer at first layer in alexnet, googlenet and so on.
@yiheng yiheng closed this Nov 21, 2016
wzhongyuan pushed a commit to wzhongyuan/BigDL that referenced this pull request Jan 16, 2019
* Add opencv support

* Add opencv support
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

7 participants