| 510 | } |
| 511 | |
| 512 | void KFAC::do_batch_begin_cbs(model& model) |
| 513 | { |
| 514 | SGDExecutionContext& c = |
| 515 | static_cast<SGDExecutionContext&>(model.get_execution_context()); |
| 516 | for (const auto& cb : model.get_callbacks()) { |
| 517 | if (c.get_step() % cb->get_batch_interval() == 0) { |
| 518 | cb->on_batch_begin(&model); |
| 519 | } |
| 520 | } |
| 521 | } |
| 522 | |
| 523 | void KFAC::do_batch_end_cbs(model& model) |
| 524 | { |
nothing calls this directly
no test coverage detected