datasets>=2.15
torch>=2.1
torchinfo>=1.8.0
torchvision>=0.16
tqdm>=4.66
