def after_iteration(self, model, epoch: int, evals_log: Dict):
if get_rabit_rank() == 0:
put_queue(lambda: self._create_checkpoint(
model, epoch, self._filename, self._frequency))
def after_iteration(self, model, epoch: int, evals_log: Dict):
if get_rabit_rank() == 0:
report_dict = self._get_report_dict(evals_log)
put_queue(lambda: tune.report(**report_dict))
def after_iteration(self, model, epoch: int, evals_log: Dict):
if get_rabit_rank() == 0:
self._checkpoint.after_iteration(model, epoch, evals_log)
self._report.after_iteration(model, epoch, evals_log)