Genkanwall commited on
Commit
8182bf8
·
1 Parent(s): 0bee6ba

Add torchtools

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -14,3 +14,4 @@ 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
 
 
14
  omegaconf>=2.3.0
15
  matplotlib>=3.7.0
16
  git+https://github.com/ildoonet/pytorch-gradual-warmup-lr.git
17
+ torchtools>=0.1.0