albumentations>=0.5.1
torch>=1.6.0
hydra-core>=1.0
timm>=0.3.2
segmentation-models-pytorch>=0.1.3
tqdm
click
colorama
tensorboard

[tests]
pytest
