refactor: disable pylint, has_overflow

kozistr · Nov 15, 2021 · 333608f · 333608f
1 parent c92f236
commit 333608f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_optimizer/utils.py b/pytorch_optimizer/utils.py
@@ -13,7 +13,7 @@ def is_valid_parameters(parameters: PARAMETERS) -> bool:
 
 def has_overflow(grad_norm: torch.Tensor) -> bool:
     """Detect inf and NaN in grad_norm."""
-    return grad_norm == float('inf') or grad_norm != grad_norm
+    return grad_norm == float('inf') or grad_norm != grad_norm  # pylint: disable=comparison-with-itself
 
 
 def normalize_gradient(x: torch.Tensor, use_channels: bool = False, epsilon: float = 1e-8) -> torch.Tensor: