ray.train.lightning.RayTrainReportCallback

class ray.train.lightning.RayTrainReportCallback(*args: Any, **kwargs: Any)

Bases: Callback

A simple callback that reports checkpoints to Ray on train epoch end.

This callback is a subclass of lightning.pytorch.callbacks.Callback.

It fetches the latest trainer.callback_metrics and reports together with the checkpoint on each training epoch end.

Checkpoints will be saved in the following structure:

checkpoint_00000*/      Ray Train Checkpoint
└─ checkpoint.ckpt      PyTorch Lightning Checkpoint

For customized reporting and checkpointing logic, implement your own lightning.pytorch.callbacks.Callback following this user guide: Saving and Loading Checkpoints.

PublicAPI (beta): This API is in beta and may change before becoming stable.

Methods

Attributes

CHECKPOINT_NAME