update: raise ValueError

Author: kozistr

pytorch_optimizer/optimizer/adai.py

@@ -125,7 +125,7 @@ def step(self, closure: CLOSURE = None) -> LOSS:
                 exp_avg_sq_hat_sum += exp_avg_sq.sum() / bias_correction2
 
         if param_size == 0:
-            raise ZeroDivisionError('[-] param_size is 0')
+            raise ValueError('[-] param_size is 0')
 
         exp_avg_sq_hat_mean = exp_avg_sq_hat_sum / param_size
 

pytorch_optimizer/optimizer/ranger21.py

@@ -236,7 +236,7 @@ def step(self, closure: CLOSURE = None) -> LOSS:
 
         # stable weight decay
         if param_size == 0:
-            raise ZeroDivisionError('[-] param_size is 0')
+            raise ValueError('[-] param_size is 0')
 
         variance_normalized = math.sqrt(variance_ma_sum / param_size)
         if math.isnan(variance_normalized):