Function get_objective_function

src/callbacks/progress_bar.cpp:96–134

}

static inline std::string get_objective_function(lbann::model* m)
{
  std::stringstream stream;
  stream << "Objective: ";

  auto terms = m->get_objective_function()->get_terms();
  bool first = true;
  for (const auto& term : terms) {
    // Only consider layer terms
    auto lterm = dynamic_cast<lbann::layer_term*>(term);
    if (!lterm)
      continue;
    lbann::Layer* layer = &lterm->get_layer();

    // Try as an EvalType evaluation layer
    lbann::EvalType objective = lbann::EvalType(-999.0);
    auto eval_layer =
      dynamic_cast<lbann::abstract_evaluation_layer<lbann::EvalType>*>(layer);
    // If not working, try as a DataType layer
    if (!eval_layer) {
      auto eval_layer_data =
        dynamic_cast<lbann::abstract_evaluation_layer<lbann::DataType>*>(layer);
      if (!eval_layer_data)
        continue;
      objective = static_cast<lbann::EvalType>(eval_layer_data->get_value());
    }
    else {
      objective = eval_layer->get_value();
    }

    if (!first)
      stream << ", ";
    stream << std::fixed << std::setprecision(4) << objective;
    first = false;
  }

  stream << " ";
  return stream.str();
}

namespace lbann {
namespace callback {