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

Fix KerasUpsampling2D 

Signed-off-by: brian <brian@brutex.de>
master
Brian Rosenberger 2022-09-21 14:42:42 +02:00
parent 978332f692
commit 597abdf063
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cavis-dnn/cavis-dnn-modelimport/src/main/java/org/deeplearning4j/nn/modelimport/keras/layers/convolutional/KerasUpsampling2D.java
View File

@ -61,14 +61,10 @@ public class KerasUpsampling2D extends KerasLayer {
super(layerConfig, enforceTrainingConfig); super(layerConfig, enforceTrainingConfig);
int[] size = KerasConvolutionUtils.getUpsamplingSizeFromConfig(layerConfig, 2, conf); int[] size = KerasConvolutionUtils.getUpsamplingSizeFromConfig(layerConfig, 2, conf);
if (size[0] != size[1])
throw new UnsupportedKerasConfigurationException("First and second size arguments have to be the same" +
"got: " + size[0] + " and " + size[1]);
Upsampling2D.Builder builder = new Upsampling2D.Builder() Upsampling2D.Builder builder = new Upsampling2D.Builder()
.name(this.layerName) .name(this.layerName)
.dropOut(this.dropout) .dropOut(this.dropout)
.size(size[0]); .size(size);
this.layer = builder.build(); this.layer = builder.build();
this.vertex = null; this.vertex = null;