MPI Model Averaging. Added check to short-circuit empty minibatch
This commit is contained in:
Родитель
3e68d18ad8
Коммит
eafc86daab
|
@ -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());
|
||||||
|
|
Загрузка…
Ссылка в новой задаче