pip>=21.0
numpy==1.19.5
torch==1.7.1
torchvision==0.8.2
torch-optimizer==0.1.0
pytorch-lightning==1.1.7
scipy==1.6.0
scikit-learn==0.24.1
kornia==0.4.1
h5py>=2.10.0
matplotlib==3.3.4
wandb==0.10.17
hydra-core==1.0.6
hydra-colorlog==1.0.0
hydra-submitit-launcher==1.1.0
