Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small fix (#233) 

Signed-off-by: Alex Black <blacka101@gmail.com>
master
Alex Black 2019-09-04 19:11:17 +10:00 committed by GitHub
parent e9454b8882
commit d41018751b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/layers/mkldnn/MKLDNNSubsamplingHelper.java
View File

@ -87,11 +87,7 @@ public class MKLDNNSubsamplingHelper implements SubsamplingHelper {
break; break;
} }
Pooling2DDerivative d = Pooling2DDerivative.derivativeBuilder() Pooling2DDerivative d = new Pooling2DDerivative(input, epsilon, gradAtInput, conf);
.config(conf)
.arrayInputs(new INDArray[]{input, epsilon})
.arrayOutputs(new INDArray[]{gradAtInput})
.build();
Nd4j.exec(d); Nd4j.exec(d);
return new Pair<Gradient,INDArray>(new DefaultGradient(), gradAtInput); return new Pair<Gradient,INDArray>(new DefaultGradient(), gradAtInput);