Update readme.md

SteveImmanuel · Jul 8, 2024 · bef1ea0 · bef1ea0
1 parent 4e49e5a
commit bef1ea0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/readme.md b/readme.md
@@ -1,5 +1,6 @@
 # Torch Cosine Annealing
 ![PyPI version](https://img.shields.io/pypi/v/torch-cosine-annealing)
+[![Downloads](https://static.pepy.tech/badge/torch-cosine-annealing)](https://pepy.tech/project/torch-cosine-annealing)
 ![Build Status](https://img.shields.io/github/actions/workflow/status/SteveImmanuel/torch-cosine-annealing/test-and-deploy.yml)
 
 Implementation of cosine annealing scheduler introduced in [SGDR](https://arxiv.org/abs/1608.03983) paper. Compared to the [original](https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.CosineAnnealingWarmRestarts.html) implementation, it has the following additional features: