torch==1.9.0
torchvision==0.10.0
PyYAML==6.0
loralib==0.1.2
easydict==1.9
tensorboard==2.8.0
opencv-python-headless==4.5.3.56
albumentations==0.5.2