apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #1429 from zoq/bigbatch_sgd_batch_bound_fix

Browse Source 
BigBatchSGD batch size fix.
This commit is contained in:
Marcus Edel
2018-06-11 19:05:35 +02:00
committed by GitHub
parent 1959b50ccb 196e6cd246
commit 0128ef7194
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/core/optimizers/bigbatch_sgd/bigbatch_sgd_impl.hpp
+1 -1
View File
@@ -143,7 +143,7 @@ double BigBatchSGD<UpdatePolicyType>::Optimize(
if (batchOffset <= 0)
batchOffset = 1;
if (batchSize + batchOffset >= numFunctions)
if ((currentFunction + batchSize + batchOffset) >= numFunctions)
break;
// Update the stochastic gradient estimation.