torch==1.7.0
torchvision==0.8.1
tqdm==4.47.0
Pillow==7.2.0
numpy==1.18.5
matplotlib==3.2.2
yaml==5.3.1
tensorboardx==2.1
tensorflow==2.15.0
tensorflow_datasets==4.9.4

