From a61e925cef6e722f31baf0cfcbc1a422844259a3 Mon Sep 17 00:00:00 2001 From: vmoens Date: Sat, 2 Dec 2023 13:09:58 +0000 Subject: [PATCH] amend --- torchrl/modules/tensordict_module/rnn.py | 1 - 1 file changed, 1 deletion(-) diff --git a/torchrl/modules/tensordict_module/rnn.py b/torchrl/modules/tensordict_module/rnn.py index 04b632e683e..ea877c31fe4 100644 --- a/torchrl/modules/tensordict_module/rnn.py +++ b/torchrl/modules/tensordict_module/rnn.py @@ -282,7 +282,6 @@ def _lstm(self, x, hx): return outputs, (torch.stack(h_t, 0), torch.stack(c_t, 0)) def forward(self, input, hx=None): # noqa: F811 - self._update_flat_weights() real_hidden_size = self.proj_size if self.proj_size > 0 else self.hidden_size if input.dim() != 3: raise ValueError(