from ..base import BaseModel


class AdamParams(BaseModel):
    learning_rate: float = 0.0001
    """学习率"""

    beta1: float = 0.9
    """AdamW beta1 (通常接近 0.9)"""

    beta2: float = 0.95
    """AdamW beta2 (通常接近 0.999)"""

    eps: float = 1e-12
    """AdamW epsilon (通常接近 1e-8)"""

    weight_decay: float = 0.0
    """权重衰减系数"""
