amend

pytorch · Dec 2, 2023 · 7e58cd8 · 7e58cd8
1 parent a61e925
commit 7e58cd8
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
@@ -966,7 +966,6 @@ def _gru(self, x, hx):
         return outputs, torch.stack(h_t, 0)
 
     def forward(self, input, hx=None):  # noqa: F811
-        self._update_flat_weights()
         if input.dim() != 3:
             raise ValueError(
                 f"GRU: Expected input to be 3D, got {input.dim()}D instead"