pepy.tech

torch-cosine-annealing

0.1.3
2.47k

Cosine annealing learning rate scheduler for PyTorch based on SGDR