numpy<1.25.2,>=1.21.0
torch>=1.13.0
matplotlib>3.5.0
torchvision>=0.14.0
tqdm>=4.62.3
torchinfo>=1.5.0
pyyaml
patchify
einops
