albumentations>=0.5.1
pytorch-lightning<1.2,>=1.1.8
torch>=1.6.0
hydra-core>=1.0
timm==0.3.2
segmentation-models-pytorch>=0.1.3
tqdm
click
colorama
tensorboard
ruamel.yaml

[tests]
pytest
