(bug fix in 1-bit workaround)

Codex-NG · Sep 17, 2016 · 203df46 · 203df46
1 parent 1dfdb14
commit 203df46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,