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