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