From 43128e708abfd4bacb4585bd5587630e44468fe9 Mon Sep 17 00:00:00 2001
From: Leon Chen <leon@md.ai>
Date: Wed, 5 Oct 2016 21:39:40 -0400
Subject: [PATCH] properly handle batch_input_shape attribute in InputLayer

---
 src/layers/InputLayer.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/layers/InputLayer.js b/src/layers/InputLayer.js
index 69f1384..017c04e 100644
--- a/src/layers/InputLayer.js
+++ b/src/layers/InputLayer.js
@@ -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
   }