numpy<2
matplotlib
torch<2.6,>=1.10.0
torchvision<=0.20.1,>=0.8.1
torchmetrics<0.8.0
pytorch-lightning==1.4.5
tensorboard
tqdm
imageio
imageio-ffmpeg
torchinfo
pytest
setuptools<=80
