matplotlib==3.1.3
numpy==1.19.5
seaborn==0.10.0
torch==1.9.0+cu111
tqdm==4.48.2