diff --git a/requirements.txt b/requirements.txt index b4f6719..a86b894 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ sklearn -torch==1.0 +torch==2.2.0 numpy