ray.air.integrations.mlflow.MLflowLoggerCallback.on_checkpoint

MLflowLoggerCallback.on_checkpoint(iteration: int, trials: List[Trial], trial: Trial, checkpoint: ray.tune.Checkpoint, **info)

Called after a trial saved a checkpoint with Tune.

Parameters:

• iteration – Number of iterations of the tuning loop.

• trials – List of trials.

• trial – Trial that just has errored.

• checkpoint – Checkpoint object that has been saved by the trial.

• **info – Kwargs dict for forward compatibility.