update: validate_beta0

Author: kozistr

pytorch_optimizer/base_optimizer.py

@@ -13,8 +13,8 @@ def validate_learning_rate(learning_rate: float):
 
     @staticmethod
     def validate_beta0(beta_0: float):
-        if not 0.0 <= beta_0 < 1.0:
-            raise ValueError(f'[-] beta0 {beta_0} must be in the range [0, 1)')
+        if not 0.0 <= beta_0 <= 1.0:
+            raise ValueError(f'[-] beta0 {beta_0} must be in the range [0, 1]')
 
     @staticmethod
     def validate_betas(betas: BETAS):