ray.rllib.core.learner.learner.Learner.restore_from_path#

Learner.restore_from_path(path: str | Path, *, component: str | None = None, **kwargs) None#

从给定的路径恢复实现类的状态。

如果提供了 component 参数,path 指的是 self 的子组件的检查点,从而允许用户只将子组件的状态加载到 self 中,而不影响任何其他状态信息(例如,只将神经网络状态加载到包含该神经网络的 Checkpointable 中,但该 Checkpointable 还包含不应通过调用此方法更改的其他状态信息)。

给定的 path 应具有以下结构并包含以下文件:

path/
    [component1]/
        [component1 subcomponentA]/
            ...
        [component1 subcomponentB]/
            ...
    [component2]/
            ...
    [cls.METADATA_FILE_NAME] (json)
    [cls.STATE_FILE_NAME] (pkl)

请注意,self.METADATA_FILE_NAME 文件不是恢复状态所必需的。

参数:
  • path – 加载实现类状态的路径,或仅加载实现类的一个子组件状态的路径(如果提供了 component)。

  • component – 如果提供了 path,则将其解释为仅子组件的检查点路径,因此仅恢复/加载该子组件的状态。在这种情况下,self 的所有其他状态保持不变。

  • **kwargs – 向前兼容的关键字参数。