amend

facebookresearch · Jul 31, 2024 · 949f76d · 949f76d
1 parent 716350e
commit 949f76d
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/benchmarl/models/gru.py b/benchmarl/models/gru.py
@@ -455,7 +455,6 @@ def _forward(self, tensordict: TensorDictBase) -> TensorDictBase:
                 output = output[..., 0, :]
         else:  # Is a global input, this is a critic
             # Check input
-            assert training and self.is_critic
             batch = input.shape[0]
             seq = input.shape[1]
             assert input.shape == (batch, seq, self.input_features)