supervised_layout_benchmark/requirements.txt

tqdm==4.42.1
scipy==1.4.1
pytest==5.3.5
numpy==1.18.1
matplotlib==3.1.3
ConfigArgParse==1.2.3
pytorch_lightning==1.1.2
PyYAML==5.3.1
scikit_learn==0.23.2
torch>=1.5.0
torchvision==0.8.1