| 155 | } |
| 156 | } |
| 157 | void debug::on_backward_prop_begin(model* m, Layer* l) |
| 158 | { |
| 159 | const auto& c = m->get_execution_context(); |
| 160 | if (m_modes.empty() || m_modes.count(c.get_execution_mode()) > 0) { |
| 161 | std::stringstream msg; |
| 162 | msg << rank_string(*m->get_comm()) << ": " << layer_string(*l) |
| 163 | << " is starting backward prop for " << batch_step_string(*m) |
| 164 | << std::endl; |
| 165 | std::cerr << msg.str(); |
| 166 | } |
| 167 | } |
| 168 | void debug::on_backward_prop_end(model* m, Layer* l) |
| 169 | { |
| 170 | const auto& c = m->get_execution_context(); |
no test coverage detected