Genkanwall commited on
Commit ·
0bee6ba
1
Parent(s): 04fb07c
Fix warmup_scheduler package
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
|
@@ -13,4 +13,4 @@ opencv-python>=4.8.0
|
|
| 13 |
einops>=0.7.0
|
| 14 |
omegaconf>=2.3.0
|
| 15 |
matplotlib>=3.7.0
|
| 16 |
-
pytorch-warmup
|
|
|
|
| 13 |
einops>=0.7.0
|
| 14 |
omegaconf>=2.3.0
|
| 15 |
matplotlib>=3.7.0
|
| 16 |
+
git+https://github.com/ildoonet/pytorch-gradual-warmup-lr.git
|