torch>=1.8.1
numpy>=1.19.2
tqdm==4.61.0
