matplotlib==3.3.4
numpy==1.21.1
torch==1.9.0+cu111
tqdm==4.61.2
