diff --git a/requirements.txt b/requirements.txt index be44ffc..61e9720 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ # --------- pytorch --------- # -f https://download.pytorch.org/whl/torch_stable.html -torch==1.10.1+cpu +torch==1.12.1 torchvision==0.11.2+cpu timm==0.6.7 pytorch-lightning==1.7.1