numpy==1.21.5
argparse==1.1
matplotlib==3.5.1
h5py==3.6.0
torch==1.12.0+cu116
