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

fix parameter sharing bug

This commit is contained in:
kaituoxu 2018-08-06 19:00:33 +09:00
Родитель d2b49543eb
Коммит e5006a4866
1 изменённых файлов: 3 добавлений и 2 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

5
Source/ComputationNetworkLib/ComputationNetwork.cpp
Просмотреть файл

@ -550,7 +550,9 @@ void ComputationNetwork::CollectInputAndLearnableParametersRec(const Computation
{ {
if (visited.find(node) != visited.end()) // allready got this one if (visited.find(node) != visited.end()) // allready got this one
return; return;
else if (node->OperationName() == OperationNameOf(InputValue) || node->OperationName() == OperationNameOf(SparseInputValue))
visited.insert(node);
if (node->OperationName() == OperationNameOf(InputValue) || node->OperationName() == OperationNameOf(SparseInputValue))
inputs.push_back(node); inputs.push_back(node);
else if (node->OperationName() == OperationNameOf(LearnableParameter) && node->IsParameterUpdateRequired()) else if (node->OperationName() == OperationNameOf(LearnableParameter) && node->IsParameterUpdateRequired())
learnableParameters.push_back(node); learnableParameters.push_back(node);
@ -561,7 +563,6 @@ void ComputationNetwork::CollectInputAndLearnableParametersRec(const Computation
if (pcnode && pcnode->HasComputed()) if (pcnode && pcnode->HasComputed())
return; return;
// recurse // recurse
visited.insert(node);
for (const auto & input : node->GetInputs()) for (const auto & input : node->GetInputs())
CollectInputAndLearnableParametersRec(input, visited, inputs, learnableParameters); CollectInputAndLearnableParametersRec(input, visited, inputs, learnableParameters);
} }