ray.train.lightning.RayTrainReportCallback#

class ray.train.lightning.RayTrainReportCallback(*args: Any, **kwargs: Any)[源代码]#

基类：Callback

一个简单的回调函数，在训练周期结束时向Ray报告检查点。

此回调是 lightning.pytorch.callbacks.Callback 的子类。

它在每个训练周期结束时获取最新的 trainer.callback_metrics 并连同检查点一起报告。

检查点将按以下结构保存:

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

对于自定义的报告和检查点逻辑，请按照此用户指南实现您自己的 lightning.pytorch.callbacks.Callback： 保存和加载检查点。

PublicAPI (测试版): 此API目前处于测试阶段，在成为稳定版本之前可能会发生变化。

方法

属性

CHECKPOINT_NAME