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) 進行裝飾。