set_recurrent_mode

class torchrl.modules.set_recurrent_mode(mode: bool | Literal['recurrent', 'sequential'] | None = True)

设置 RNN 循环模式的上下文管理器。

参数:

mode (bool, "recurrent" 或 "sequential") – 在上下文管理器中使用的循环模式。 “recurrent” 导致 mode=True，而 “sequential” 导致 mode=False。以 recurrent_mode “on” 执行的 RNN 假定数据以时间批次的形式传入，否则假定 tensordict 中的每个数据元素彼此独立。此上下文管理器的默认值为 True。默认的循环模式为 None，即使用 RNN 的默认循环模式（请参阅 LSTMModule 和 GRUModule 构造函数）。

另请参阅

recurrent_mode`.

注意

默认情况下，TorchRL 的所有方法都用 set_recurrent_mode(True) 进行装饰。