featup / requirements.txt
abreza's picture
update reqs
0b59808
raw
history blame
114 Bytes
torch
kornia
omegaconf
pytorch-lightning
torchvision
tqdm
torchmetrics
scikit-learn
numpy
matplotlib
timm==0.4.12