microsoft
/
CNTK
зеркало из
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

Fix bugs when loadBestModel in AdjustAfterEpoch

This commit is contained in:
Yuchen Fan 2015-11-23 15:05:43 +08:00
Родитель 99b4cae376
Коммит dea7df7e2b
1 изменённых файлов: 17 добавлений и 3 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

20
MachineLearning/CNTKSGDLib/SGD.cpp
Просмотреть файл

@ -901,10 +901,10 @@ namespace Microsoft { namespace MSR { namespace CNTK {
if (m_loadBestModel)
{
fprintf(stderr, "Loaded the previous model which has better training criterion.\n");
net->LoadPersistableParametersFromFile(GetModelNameForEpoch(i - 1),
net->LoadPersistableParametersFromFile(GetModelNameForEpoch(i - m_learnRateAdjustInterval),
m_validateAfterModelReloading);
net->ResetEvalTimeStamp();
LoadCheckPointInfo(i - 1,
LoadCheckPointInfo(i - m_learnRateAdjustInterval,
/*out*/ totalSamplesSeen,
/*out*/ learnRatePerSample,
smoothedGradients,
@ -984,7 +984,21 @@ namespace Microsoft { namespace MSR { namespace CNTK {
if (!m_keepCheckPointFiles)
{
// delete previous checkpoint file to save space
_wunlink(GetCheckPointFileNameForEpoch(i - 1).c_str());
if (m_autoLearnRateSearchType == LearningRateSearchAlgorithm::AdjustAfterEpoch && m_loadBestModel)
{
if (epochsSinceLastLearnRateAdjust != 1)
{
_wunlink(GetCheckPointFileNameForEpoch(i - 1).c_str());
}
if (epochsSinceLastLearnRateAdjust == m_learnRateAdjustInterval)
{
_wunlink(GetCheckPointFileNameForEpoch(i - m_learnRateAdjustInterval).c_str());
}
}
else
{
_wunlink(GetCheckPointFileNameForEpoch(i - 1).c_str());
}
}
}