datasets >= 2.0.0
torch >= 1.3
accelerate
evaluate
Pillow
albumentations