diff --git a/requirements.txt b/requirements.txt index 3a4f146..7ace470 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -pytorch-lightning==0.8.5 +pytorch-lightning==2.4.0 omegaconf==2.0.1 tensorboard==1.15 sklearn