fix subsampling layer layer factory type

deeplearning4j · May 18, 2015 · 6e03726 · 6e03726
1 parent 42df413
commit 6e03726
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/conf/NeuralNetConfiguration.java b/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/conf/NeuralNetConfiguration.java
@@ -787,10 +787,13 @@ public ListBuilder hiddenLayerSizes(int...hiddenLayerSizes) {
             return this;
         }
 
+        /**
+         * Build the multi layer network
+         * based on this neural network and
+         * overr ridden parameters
+         * @return the configuration to build
+         */
         public MultiLayerConfiguration build() {
-            if(layerwise.size() != hiddenLayerSizes.length + 1)
-                throw new IllegalStateException("Number of hidden layers mut be equal to hidden layer sizes + 1");
-
             List<NeuralNetConfiguration> list = new ArrayList<>();
             for(int i = 0; i < layerwise.size(); i++) {
                 if(confOverrides.get(i) != null)

diff --git a/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/layers/factory/LayerFactories.java b/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/layers/factory/LayerFactories.java
@@ -22,7 +22,6 @@
 import org.deeplearning4j.nn.api.LayerFactory;
 import org.deeplearning4j.nn.conf.NeuralNetConfiguration;
 import org.deeplearning4j.nn.conf.layers.*;
-import org.deeplearning4j.nn.layers.convolution.subsampling.SubsamplingLayer;
 
 /**
  * Static method for finding which layer factory to use