Skip to content

Commit d6295a3

Browse files
author
Heaton
committed
Batch size was not initially set for SGD
1 parent 752402a commit d6295a3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/main/java/org/encog/neural/networks/training/propagation/sgd/StochasticGradientDescent.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -314,6 +314,7 @@ public void setBatchSize(int theBatchSize) {
314314
((BatchDataSet)getTraining()).setBatchSize(theBatchSize);
315315
} else {
316316
BatchDataSet batchSet = new BatchDataSet(getTraining(),this.rnd);
317+
batchSet.setBatchSize(theBatchSize);
317318
setTraining(batchSet);
318319
}
319320
}

0 commit comments

Comments
 (0)