numpy<2
matplotlib
torch<=1.12.1,>=1.10.0
torchvision>=0.8.1
torchmetrics<0.8.0
pytorch-lightning==1.4.5
tensorboard<2.12
tqdm
imageio
imageio-ffmpeg
torchinfo
pytest
