torch>=2.2.1
tqdm
numpy
