numpy<1.25.2,>=1.21.0
torch<=2.0.1,>=1.13.0
torchvision<=0.15.0,>=0.14.0
tqdm<=4.65.0,>=4.62.3
