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

Simple fix in bidirectional lstm import (#293)

Browse Source 
* first pass

Signed-off-by: eraly <susan.eraly@gmail.com>

* cleanup

Signed-off-by: eraly <susan.eraly@gmail.com>
This commit is contained in:
Susan Eraly 2020-03-30 03:07:47 -07:00 committed by GitHub
parent 63c9223bc2
commit 3900d9ff06
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deeplearning4j/deeplearning4j-modelimport/src/main/java/org/deeplearning4j/nn/modelimport/keras/layers/wrappers/KerasBidirectional.java
View File

@ -190,7 +190,7 @@ public class KerasBidirectional extends KerasLayer {
"Keras Bidirectional layer accepts only one input (received " + inputType.length + ")");
InputPreProcessor preProcessor = getInputPreprocessor(inputType);
if (preProcessor != null)
return preProcessor.getOutputType(inputType[0]);
return this.getBidirectionalLayer().getOutputType(-1, preProcessor.getOutputType(inputType[0]));
else
return this.getBidirectionalLayer().getOutputType(-1, inputType[0]);
}