amend

pytorch · Dec 2, 2023 · a61e925 · a61e925
1 parent c3a92aa
commit a61e925
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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(