examples/mnist_example2.rb in ruby-dnn-0.2.2 vs examples/mnist_example2.rb in ruby-dnn-0.3.0

@@ -23,16 +23,16 @@
 
 model = Model.new
 
 model << InputLayer.new([28, 28, 1])
 
-model << Conv2D.new(16, 5, 5)
+model << Conv2D.new(16, 5)
 model << BatchNormalization.new
 model << ReLU.new
 
-model << MaxPool2D.new(2, 2)
+model << MaxPool2D.new(2)
 
-model << Conv2D.new(32, 5, 5)
+model << Conv2D.new(32, 5)
 model << BatchNormalization.new
 model << ReLU.new
 
 model << Flatten.new