From 8e591bbf39b60c6fad74d2661ec9acbb055eddc3 Mon Sep 17 00:00:00 2001 From: Rhys Compton Date: Mon, 23 Nov 2020 21:07:40 +1300 Subject: [PATCH] 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. * :bug: 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 --- .../src/main/java/org/deeplearning4j/zoo/model/LeNet.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeplearning4j/deeplearning4j-zoo/src/main/java/org/deeplearning4j/zoo/model/LeNet.java b/deeplearning4j/deeplearning4j-zoo/src/main/java/org/deeplearning4j/zoo/model/LeNet.java index 4f679b85f..8a59741b1 100644 --- a/deeplearning4j/deeplearning4j-zoo/src/main/java/org/deeplearning4j/zoo/model/LeNet.java +++ b/deeplearning4j/deeplearning4j-zoo/src/main/java/org/deeplearning4j/zoo/model/LeNet.java @@ -55,7 +55,7 @@ import org.nd4j.linalg.lossfunctions.LossFunctions; public class LeNet extends ZooModel { @Builder.Default private long seed = 1234; - @Builder.Default private int[] inputShape = new int[] {3, 224, 224}; + @Builder.Default private int[] inputShape = new int[] {1, 28, 28}; @Builder.Default private int numClasses = 0; @Builder.Default private IUpdater updater = new AdaDelta(); @Builder.Default private CacheMode cacheMode = CacheMode.NONE;