Genkanwall commited on
Commit
04fb07c
·
1 Parent(s): 854c65a

Add warmup_scheduler

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -13,3 +13,4 @@ opencv-python>=4.8.0
13
  einops>=0.7.0
14
  omegaconf>=2.3.0
15
  matplotlib>=3.7.0
 
 
13
  einops>=0.7.0
14
  omegaconf>=2.3.0
15
  matplotlib>=3.7.0
16
+ pytorch-warmup>=0.1.1