example/example_mnist.rb in menoh-1.0.0 vs example/example_mnist.rb in menoh-1.0.1
- old
+ new
@@ -34,11 +34,11 @@
{
name: MNIST_IN_NAME,
dims: [
image_list.length,
input_shape[:channel_num],
+ input_shape[:height],
input_shape[:width],
- input_shape[:height]
]
}
],
output_layers: [MNIST_OUT_NAME]
}