Method forward_prop

src/models/model.cpp:1553–1582

}

void model::forward_prop(execution_mode mode)
{
  LBANN_CALIPER_MARK_FUNCTION;
  do_model_forward_prop_begin_cbs(mode);

  // Clear activations in reference counter
  m_activation_refcnt.clear();

  for (El::Int i = 0; i < get_num_layers(); ++i) {
    auto& l = get_layer(i);

    if (this->is_subgraph_parallelism_enabled()) {
      if (l.get_run_layer_in_subgraph() || l.get_name() == "layer1") {
        do_layer_forward_prop_begin_cbs(mode, &l);
        l.forward_prop();
        do_layer_forward_prop_end_cbs(mode, &l);
      }
      else {
        // To Do: Fix last batch problem in sub-graph parallelism
        // experimental code to fix last batch problem in subgraph parallelism
      }
    }
    else {
      do_layer_forward_prop_begin_cbs(mode, &l);
      l.forward_prop();
      do_layer_forward_prop_end_cbs(mode, &l);
    }
  }
  do_model_forward_prop_end_cbs(mode);
}

void model::backward_prop(bool compute_weight_grads_only)
{