Skip to content
Snippets Groups Projects
Commit 0c512256 authored by Van Dung NGUYEN's avatar Van Dung NGUYEN
Browse files

add more info

parent 1da4e955
No related branches found
No related tags found
1 merge request!309Master
...@@ -1739,6 +1739,7 @@ void TrainingDeepMaterialNetwork::train(double lr, int maxEpoch, std::string los ...@@ -1739,6 +1739,7 @@ void TrainingDeepMaterialNetwork::train(double lr, int maxEpoch, std::string los
int Ntrain = _XTrain.size(); int Ntrain = _XTrain.size();
double lrCur = lr; double lrCur = lr;
int numEpochIncrease = 0; int numEpochIncrease = 0;
int numEpochLrMin = 0;
int numEpochDecrease = 0; int numEpochDecrease = 0;
while (true) while (true)
{ {
...@@ -1818,6 +1819,12 @@ void TrainingDeepMaterialNetwork::train(double lr, int maxEpoch, std::string los ...@@ -1818,6 +1819,12 @@ void TrainingDeepMaterialNetwork::train(double lr, int maxEpoch, std::string los
if (lrCur < lrmin) if (lrCur < lrmin)
{ {
lrCur = lrmin; lrCur = lrmin;
numEpochLrMin ++;
if (numEpochLrMin>10)
{
Msg::Info("maximal number of iterations %d at minimal learning step is reached !!",numEpochLrMin);
break;
}
} }
numEpochDecrease = 0; numEpochDecrease = 0;
Wcur = Wprev; Wcur = Wprev;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment