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

properly handle batch_input_shape attribute in InputLayer

Browse Source
This commit is contained in:
Leon Chen
2016-10-05 21:39:40 -04:00
parent f5e6e685db
commit 43128e708a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/layers/InputLayer.js
+2 -2
View File
@@ -16,8 +16,8 @@ export default class InputLayer extends Layer {
shape = []
} = attrs
this.shape = attrs.batch_input_shape && attrs.batch_input_shape.length
? attrs.batch_input_shape.slice(1)
this.shape = attrs.batchInputShape && attrs.batchInputShape.length
? attrs.batchInputShape.slice(1)
: shape
}