omegaconf>=2.1
numpy>=1.20.0
torch>=2.3.1
torchvision>=0.18.1
lightning>=2.0
six>=1.9
einops>=0.4.0
