ray.rllib.utils.checkpoints.Checkpointable.get_state#
- abstract Checkpointable.get_state(components: str | Collection[str] | None = None, *, not_components: str | Collection[str] | None = None, **kwargs) Dict[str, Any] [源代码]#
返回实现类的当前状态为字典。
- 参数:
components – 一个可选的字符串键集合,包含在返回的状态中。如果获取状态的某些部分开销很大(例如,读取/编译一个大型神经网络的权重),同时这些部分对调用者来说不是必需的,这可能会很有用。
not_components – 一个可选的字符串键列表,即使在
components
中包含相同的字符串,也要在返回的状态中排除。这对于获取类的完整状态非常有用,除了一个或几个组件。kwargs – 向前兼容的关键字参数。
- 返回:
实现类的当前状态(或仅指定
components
,不包括not_components
中的组件)。