Fixed builder formatting
This commit is contained in:
parent
c67658322b
commit
768658cfc8
|
@ -30,23 +30,23 @@ class CnnModel {
|
||||||
this.properties = properties;
|
this.properties = properties;
|
||||||
|
|
||||||
MultiLayerConfiguration configuration = new NeuralNetConfiguration.Builder()
|
MultiLayerConfiguration configuration = new NeuralNetConfiguration.Builder()
|
||||||
.seed(1611)
|
.seed(1611)
|
||||||
.optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)
|
.optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)
|
||||||
.learningRate(properties.getLearningRate())
|
.learningRate(properties.getLearningRate())
|
||||||
.regularization(true)
|
.regularization(true)
|
||||||
.updater(properties.getOptimizer())
|
.updater(properties.getOptimizer())
|
||||||
.list()
|
.list()
|
||||||
.layer(0, conv5x5())
|
.layer(0, conv5x5())
|
||||||
.layer(1, pooling2x2Stride2())
|
.layer(1, pooling2x2Stride2())
|
||||||
.layer(2, conv3x3Stride1Padding2())
|
.layer(2, conv3x3Stride1Padding2())
|
||||||
.layer(3, pooling2x2Stride1())
|
.layer(3, pooling2x2Stride1())
|
||||||
.layer(4, conv3x3Stride1Padding1())
|
.layer(4, conv3x3Stride1Padding1())
|
||||||
.layer(5, pooling2x2Stride1())
|
.layer(5, pooling2x2Stride1())
|
||||||
.layer(6, dense())
|
.layer(6, dense())
|
||||||
.pretrain(false)
|
.pretrain(false)
|
||||||
.backprop(true)
|
.backprop(true)
|
||||||
.setInputType(dataSetService.inputType())
|
.setInputType(dataSetService.inputType())
|
||||||
.build();
|
.build();
|
||||||
|
|
||||||
network = new MultiLayerNetwork(configuration);
|
network = new MultiLayerNetwork(configuration);
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue