Merge pull request #55 from xqtbox/master

谢谢更改
MorvanZhou · Aug 31, 2017 · 50e2352 · 50e2352
2 parents f94533e + bfb23d0
commit 50e2352
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kerasTUT/6-CNN_example.py b/kerasTUT/6-CNN_example.py
@@ -37,7 +37,7 @@
 
 # Conv layer 1 output shape (32, 28, 28)
 model.add(Convolution2D(
-    batch_input_shape=(64, 1, 28, 28),
+    batch_input_shape=(None, 1, 28, 28),
     filters=32,
     kernel_size=5,
     strides=1,