From 0c5122560f95dbe2f54e97cca7fc080b72e41151 Mon Sep 17 00:00:00 2001
From: Van Dung Nguyen <vdg.nguyen@gmail.com>
Date: Wed, 28 Oct 2020 13:51:29 +0100
Subject: [PATCH] add more info
---
NonLinearSolver/modelReduction/DeepMaterialNetworks.cpp | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/NonLinearSolver/modelReduction/DeepMaterialNetworks.cpp b/NonLinearSolver/modelReduction/DeepMaterialNetworks.cpp
index f49a291bc..1fdead313 100644
--- a/NonLinearSolver/modelReduction/DeepMaterialNetworks.cpp
+++ b/NonLinearSolver/modelReduction/DeepMaterialNetworks.cpp
@@ -1739,6 +1739,7 @@ void TrainingDeepMaterialNetwork::train(double lr, int maxEpoch, std::string los
int Ntrain = _XTrain.size();
double lrCur = lr;
int numEpochIncrease = 0;
+ int numEpochLrMin = 0;
int numEpochDecrease = 0;
while (true)
{
@@ -1818,6 +1819,12 @@ void TrainingDeepMaterialNetwork::train(double lr, int maxEpoch, std::string los
if (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;
Wcur = Wprev;
--
GitLab