python==3.6.9
pytorch==1.6.0
numpy==1.19.5
pillow==8.1.2
torchvision==0.7.0
tqdm==4.36.1
scikit-learn==0.24.1
pytz
colorama