MCPcopy Create free account
hub / github.com/LBANN/lbann / do_batch_begin_cbs

Method do_batch_begin_cbs

src/execution_algorithms/kfac.cpp:512–521  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

510}
511
512void 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
523void KFAC::do_batch_end_cbs(model& model)
524{

Callers

nothing calls this directly

Calls 4

get_stepMethod · 0.80
get_batch_intervalMethod · 0.80
get_callbacksMethod · 0.45
on_batch_beginMethod · 0.45

Tested by

no test coverage detected