This commit is contained in:
Qiwei Ye 2016-05-20 13:45:45 +08:00
Родитель fdb82ac79f
Коммит c72ca0e4c1
1 изменённых файлов: 5 добавлений и 6 удалений

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

@ -1123,18 +1123,17 @@ size_t SGD<ElemType>::TrainOneEpoch(ComputationNetworkPtr net,
// using parameter server for parameter update
if (userAsyncGradientAggregation && m_mpi->NumNodesInUse() > 1)
{
if (nSamplesSinceLastModelSync >= m_nFramesBetweenASGDSync[epochNumber])
{
m_pMultiversoHelper->PushAndPullModel(learnableNodes, nSamplesSinceLastModelSync);
nSamplesSinceLastModelSync = 0;
}
// Determine if any samples were processed across any of the ranks
if (useDistributedMBReading)
{
noMoreSamplesToProcess = !wasDataRead;
}
if (nSamplesSinceLastModelSync >= m_nFramesBetweenASGDSync[epochNumber])
{
m_pMultiversoHelper->PushAndPullModel(learnableNodes, nSamplesSinceLastModelSync);
nSamplesSinceLastModelSync = 0;
}
}
commTimer.Stop();