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

[tests]
pytest
