remove assert

facebookresearch · Nov 8, 2023 · 1f85bf2 · 1f85bf2
1 parent acc138d
commit 1f85bf2
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/benchmarl/experiment/experiment.py b/benchmarl/experiment/experiment.py
@@ -8,7 +8,6 @@
 
 import importlib
 
-import math
 import os
 import time
 from collections import OrderedDict
@@ -641,7 +640,6 @@ def _optimizer_loop(self, group: str) -> TensorDictBase:
 
                 grad_norm = self._grad_clip(optimizer)
 
-                assert not math.isnan(grad_norm)
                 training_td.set(
                     f"grad_norm_{loss_name}",
                     torch.tensor(grad_norm, device=self.config.train_device),