matplotlib==3.5.2
numpy==1.23.1
torch==1.12.0+cu116
tqdm==4.64.0