diff --git a/src/mlpack/core/optimizers/sgd/sgd_impl.hpp b/src/mlpack/core/optimizers/sgd/sgd_impl.hpp index 9b9a6e3111..166c499a72 100644 --- a/src/mlpack/core/optimizers/sgd/sgd_impl.hpp +++ b/src/mlpack/core/optimizers/sgd/sgd_impl.hpp @@ -60,7 +60,7 @@ double SGD::Optimize(arma::mat& iterate) Log::Info << "SGD: iteration " << i << ", objective " << overallObjective << "." << std::endl; - if (std::isnan(overallObjective)) + if (std::isnan(overallObjective) || std::isinf(overallObjective)) { Log::Warn << "SGD: converged to " << overallObjective << "; terminating" << " with failure. Try a smaller step size?" << std::endl;