examples/mnist_example.rb in ruby-dnn-0.9.4 vs examples/mnist_example.rb in ruby-dnn-0.10.0

@@ -1,7 +1,8 @@
 require "dnn"
-require "dnn/lib/mnist"
+require "dnn/mnist"
+# If you use numo/linalg then please uncomment out.
 # require "numo/linalg/autoloader"
 
 include DNN::Layers
 include DNN::Activations
 include DNN::Optimizers
@@ -24,14 +25,12 @@
 model = Model.new
 
 model << InputLayer.new(784)
 
 model << Dense.new(256)
-model << BatchNormalization.new
 model << ReLU.new
 
 model << Dense.new(256)
-model << BatchNormalization.new
 model << ReLU.new
 
 model << Dense.new(10)
 
 model.compile(RMSProp.new, SoftmaxCrossEntropy.new)