numpy==1.21.4
hydra-core==1.1.1
tensorboardX==2.5
torch==1.10.1
torchvision==0.11.2
omegaconf==2.1.1
scipy==1.7.3
sklearn==0.0