MPI Model Averaging. Added check to short-circuit empty minibatch

This commit is contained in:
Jasha Droppo 2015-01-14 18:51:05 -08:00
Родитель 3e68d18ad8
Коммит eafc86daab
1 изменённых файлов: 3 добавлений и 1 удалений

Просмотреть файл

@ -1040,6 +1040,8 @@ namespace Microsoft { namespace MSR { namespace CNTK {
UpdateEvalTimeStamps(labelNodes); UpdateEvalTimeStamps(labelNodes);
size_t actualMBSize = net.GetActualMBSize(); size_t actualMBSize = net.GetActualMBSize();
if (0 == actualMBSize)
continue;
net.SetActualMiniBatchSize(actualMBSize); net.SetActualMiniBatchSize(actualMBSize);
net.SetActualNbrSlicesInEachRecIter(trainSetDataReader->NumberSlicesInEachRecurrentIter()); net.SetActualNbrSlicesInEachRecIter(trainSetDataReader->NumberSlicesInEachRecurrentIter());