ray.rllib.core.models.catalog.Catalog.build_encoder

Catalog.build_encoder(framework: str) → Encoder

构建编码器。

默认情况下，此方法从 Catalog._encoder_config 构建一个编码器实例。

如果你想使用 RLlib 的默认 RL 模块，但只想更改编码器，你应该重写这个。例如，如果你想使用自定义编码器，但想使用 RLlib 的默认头部、动作分布以及张量在这些组件之间的路由方式。如果你想完全控制 RL 模块，我们建议通过继承 RLlib 的 RL 模块之一来编写你自己的 RL 模块。

参数:

framework – 要使用的框架。可以是“torch”或“tf2”。

返回:

编码器。