diff --git a/requirements.txt b/requirements.txt index 0cd8b07..c2a6dff 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ opencv-python pudb==2019.2 imageio==2.9.0 imageio-ffmpeg==0.4.2 -pytorch-lightning==1.4.2 +pytorch-lightning==1.6.0 torchmetrics==0.6 omegaconf==2.1.1 test-tube>=0.7.5