Skip to content

Commit f925ec7

Browse files
committed
refactor: ValueError to NegativeLRError
1 parent 5b8a636 commit f925ec7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytorch_optimizer/optimizer/ranger21.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import torch.nn.functional as F
66
from torch.optim import Optimizer
77

8-
from pytorch_optimizer.base.exception import NoSparseGradientError, ZeroParameterSizeError
8+
from pytorch_optimizer.base.exception import NegativeLRError, NoSparseGradientError, ZeroParameterSizeError
99
from pytorch_optimizer.base.optimizer import BaseOptimizer
1010
from pytorch_optimizer.base.types import BETAS, CLOSURE, DEFAULTS, LOSS, PARAMETERS
1111
from pytorch_optimizer.optimizer.agc import agc
@@ -253,7 +253,7 @@ def step(self, closure: CLOSURE = None) -> LOSS:
253253
# warm down
254254
lr = self.get_warm_down(lr, step)
255255
if lr < 0.0:
256-
raise ValueError(f'{lr} went negative')
256+
raise NegativeLRError(lr)
257257

258258
# stable decay
259259
decay = group['weight_decay']

0 commit comments

Comments
 (0)