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。以迴圈模式“開啟”執行的 RNN 假設資料是以時間批次的形式傳入，否則假設 tensordict 中的每個資料元素是相互獨立的。此上下文管理器的預設值為 True。預設迴圈模式是 None，即使用 RNN 的預設迴圈模式（參見 LSTMModule 和 GRUModule 建構函式）。

另請參閱

recurrent_mode`.

注意

預設情況下，所有 TorchRL 方法都使用 set_recurrent_mode(True) 進行裝飾。