From 68a02f8a6be3fb06621d0c5a1b967576efb1aa45 Mon Sep 17 00:00:00 2001 From: kozistr Date: Sat, 1 Feb 2025 21:36:40 +0900 Subject: [PATCH] update: remove torch.jit.script decorator --- pytorch_optimizer/optimizer/psgd_utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/pytorch_optimizer/optimizer/psgd_utils.py b/pytorch_optimizer/optimizer/psgd_utils.py index 04ca7c96..b80bdb35 100644 --- a/pytorch_optimizer/optimizer/psgd_utils.py +++ b/pytorch_optimizer/optimizer/psgd_utils.py @@ -72,7 +72,6 @@ def triu_with_diagonal_and_above(a: torch.Tensor) -> torch.Tensor: return torch.triu(a, diagonal=0) + torch.triu(a, diagonal=1) -@torch.jit.script def update_precondition_dense( q: torch.Tensor, dxs: List[torch.Tensor], dgs: List[torch.Tensor], step: float = 0.01, eps: float = 1.2e-38 ) -> torch.Tensor: