torch>=1.9
matplotlib>=3.3
tqdm>=4.0
