(bug fix in 1-bit workaround)

2016-09-16 20:29:11 -07:00 · 2016-09-16 20:29:11 -07:00 · 203df465dd
--- a/Source/SGDLib/SGD.cpp
+++ b/Source/SGDLib/SGD.cpp
@ -1845,7 +1845,7 @@ size_t SGD<ElemType>::SearchForBestMinibatchSize(ComputationNetworkPtr net,
        EpochCriterion epochCriterion(EpochCriterion::Infinity());
        TrainOneMiniEpochAndReloadModel(net, refNet, refNode, epochNumber,
                                        numFramesToUseInSearch, trainSetDataReader,
-                                        learnRatePerSample, trialMinibatchSize, featureNodes,
+                                        learnRatePerSample, lastGoodMinibatchSize, featureNodes,
                                        labelNodes, criterionNodes,
                                        evaluationNodes, inputMatrices,
                                        learnableNodes, smoothedGradients, smoothedCounts,