torch>=1.6
numpy>=1.14
Pillow
comet_ml
pyyaml>=5.4
