diff --git a/experiments/src/main/java/samples/NeuralNetMnistPretrain.java b/experiments/src/main/java/samples/NeuralNetMnistPretrain.java index a013562f16..5e7bd0e8e8 100644 --- a/experiments/src/main/java/samples/NeuralNetMnistPretrain.java +++ b/experiments/src/main/java/samples/NeuralNetMnistPretrain.java @@ -86,7 +86,7 @@ protected void preTrain(Layer[] ls, int index) { Layer.shareWeights(ls[index], pre[index]); Layer.shareWeights(ls[index], pre[index + 1]); for( int i = 0; i < pre.length; i++ ) { - pre[i].init(pre, i, false, 0); + pre[i].init(pre, i, false); } _trainer = new Trainer.Direct(pre, 10, self());