Merge pull request #31 from khaledkah/khaledkah-patch-1

Update callbacks.py
khaledkah · Jul 9, 2024 · ebde24d · ebde24d
2 parents 8f84b85 + 916ae11
commit ebde24d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/morered/callbacks.py b/src/morered/callbacks.py
@@ -57,8 +57,8 @@ def save_outputs(
 
         # check if loss is exploding
         if (
-            key not in outputs or
-            (outputs[key] > self.loss_threshold).any()
+            key not in outputs
+            or (outputs[key] > self.loss_threshold).any()
             or torch.isnan(outputs[key]).any()
             or torch.isinf(outputs[key]).any()
         ):