| 144 | } |
| 145 | } |
| 146 | void debug::on_forward_prop_end(model* m, Layer* l) |
| 147 | { |
| 148 | const auto& c = m->get_execution_context(); |
| 149 | if (m_modes.empty() || m_modes.count(c.get_execution_mode()) > 0) { |
| 150 | std::stringstream msg; |
| 151 | msg << rank_string(*m->get_comm()) << ": " << layer_string(*l) |
| 152 | << " is ending forward prop for " << batch_step_string(*m) |
| 153 | << std::endl; |
| 154 | std::cerr << msg.str(); |
| 155 | } |
| 156 | } |
| 157 | void debug::on_backward_prop_begin(model* m, Layer* l) |
| 158 | { |
| 159 | const auto& c = m->get_execution_context(); |
nothing calls this directly
no test coverage detected