torchwrench.optim.schedulers module¶ class torchwrench.optim.schedulers.CosDecayRule(n_steps: int)[source]¶ Bases: object class torchwrench.optim.schedulers.CosDecayScheduler(optimizer: Optimizer, n_steps: int, last_epoch: int = -1)[source]¶ Bases: LambdaLR