cavis/deeplearning4j/deeplearning4j-zoo
Rhys Compton 8e591bbf39
Fix lenet input shape (#9130)
* Add options for changing the default CNN2D format

This means that when a model is being loaded (e.g., from a Keras H5 file), the default CNN2DFormat can be set. Currently it always defaults to 'Channels First' which can cause problems for some models.

* 🐛 Fix default Input shape for LeNet

Input shape should be [1, 28, 28], as per manual testing and https://github.com/BVLC/caffe/blob/master/examples/mnist/lenet.prototxt
2020-11-23 17:07:40 +09:00
..
src Fix lenet input shape (#9130) 2020-11-23 17:07:40 +09:00
pom.xml Development updates (#9053) 2020-07-26 21:59:27 +09:00