[onert] Update TrainingInfo::isValid() condition (Samsung#12360)

This PR update TrainingInfo::isValid() condtion to check learning_rate has positive value. ONE-DCO-1.0-Signed-off-by: SeungHui Youn <[email protected]>
mhs4670go · Dec 26, 2023 · 4d423d1 · 4d423d1
1 parent 0d3c932
commit 4d423d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/runtime/onert/core/src/ir/train/TrainingInfo.cc b/runtime/onert/core/src/ir/train/TrainingInfo.cc
@@ -31,7 +31,7 @@ bool TrainingInfo::isValid() const
   if (_optimizer_info.optim_code == onert::ir::train::OptimizerCode::Invalid)
     return false;
 
-  if (_optimizer_info.learning_rate == 0.0f)
+  if (_optimizer_info.learning_rate <= 0.0f)
     return false;
 
   if (_loss_info.loss_code == onert::ir::train::LossCode::Invalid)