diff --git a/requirements.txt b/requirements.txt index d4661a3..4b397b5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ pytorch-lightning==1.6.0 -torch==1.11 +torch==2.2.0 timm==0.5.4 torchvision==0.12.0 scikit-learn==1.0.2