requirements.txt

torch>=1.12.1
torchvision==0.13.1
pytorch-ignite==0.4.11
pyyaml
pytest
openmim==0.3.7
mmcv==2.0.0
mmpretrain==1.0.0rc7
mmsegmentation==1.0.0
timm==0.6.12
gradio==3.50.2
tqdm
cvxpylayers @ git+https://github.com/cvxgrp/cvxpylayers.git
qpth @ git+https://github.com/locuslab/qpth.git
torch_optimizer
scikit-learn