Searched refs:max_steps_ (Results 1 – 2 of 2) sorted by relevance
32 : ds_(dataset), check_every_n_(check_every_n), max_steps_(max_steps) {} in ds_()42 int max_steps_; variable
37 if ((cb_data.epoch_ + 1) % check_every_n_ == 0) cb_data.loop_->Eval(ds_, {}, nullptr, max_steps_); in EpochEnd()