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 – 向前兼容的关键字参数。