requirements.txt 237 B

1234567891011121314151617
  1. torch # for full alignment 2.2.0
  2. torchvision
  3. torchaudio
  4. packaging
  5. triton
  6. timm==0.4.12
  7. pytest
  8. chardet
  9. yacs
  10. termcolor
  11. submitit
  12. tensorboardX
  13. fvcore
  14. seaborn
  15. # timm==0.5.4 # for s4nd
  16. # hydra-core # for s4nd
  17. # pytorch_lightning # for s4nd