ray.rllib.models.modelv2.ModelV2.__init__

ModelV2.__init__(obs_space: gymnasium.spaces.Space, action_space: gymnasium.spaces.Space, num_outputs: int, model_config: dict, name: str, framework: str)

初始化一个 ModelV2 实例。

此方法应创建模型使用的任何变量。

参数:

• obs_space – 目标 gym 环境的观察空间。这可能有一个 original_space 属性，用于指定如何将张量展开为不规则张量。

• action_space – 目标gym环境的行为空间。

• num_outputs – 模型的输出单元数量。

• model_config – 模型的配置，记录在 ModelCatalog 中。

• name – 模型的名称（范围）。

• framework – 可以是“tf”或“torch”。