ray.tune.Trainable.save

Trainable.save(checkpoint_dir: str | None = None) → _TrainingResult

将当前模型状态保存到检查点。

子类应重写 save_checkpoint() 以保存状态。

参数:

checkpoint_dir – 可选的目录来放置检查点。

返回:

给定或创建的检查点目录。

注意返回值与 restore() 的预期相匹配。

开发者API: 此API可能会在Ray的次要版本之间发生变化。