wassname/keras-js
1
0
Fork 0
mirror of https://github.com/wassname/keras-js.git synced 2026-09-09 11:25:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix Deconvolution2D layer outputShape

Browse Source
This commit is contained in:
Leon Chen
2016-10-05 18:52:01 -04:00
parent a9f6a6bd0b
commit 5afee8e792
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/layers/convolutional/Deconvolution2D.js
+5 -1
View File
@@ -37,7 +37,11 @@ export default class Deconvolution2D extends Layer {
this.kernelShape = [nbFilter, nbRow, nbCol]
this.outputShape = outputShape
if (outputShape[0] == null) {
this.outputShape = outputShape.slice(1)
} else {
this.outputShape = outputShape
}
this.activation = activations[activation]