mindformers.core.lr.PolynomialWithWarmUpLR

class mindformers.core.lr.PolynomialWithWarmUpLR(learning_rate: float, warmup_steps: int, total_steps: int, lr_end: float = 1e-07, power: float = 1.0, warmup_lr_init: float = 0.0)[源代码]

PolynomialWithWarmUpLR.