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 中的组件)。