ray.train.trainer.BaseTrainer.training_loop

abstract BaseTrainer.training_loop() → None

由 fit() 调用的循环，用于运行训练并将结果报告给 Tune。

备注

此方法在远程进程上运行。

如果 self.datasets 被封装在一个工厂中，它们已经被评估过了。

你可以在训练循环中使用 Ray Train 工具 (train.report() 和 train.get_checkpoint())。

示例：

from ray.train.trainer import BaseTrainer
from ray import train

class MyTrainer(BaseTrainer):
    def training_loop(self):
        for epoch_idx in range(5):
            ...
            train.report({"epoch": epoch_idx})