diff --git a/requirements.txt b/requirements.txt index bad3482..63a3ebc 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ # -*- coding: utf-8 -*- numpy -torch==2.2.2 +torch>=2.2.2 torchvision==0.18.1 scikit-learn yacs