WARNING Change torch and torchvision version

- this has NOT been tested
2024-12-25 12:50:22 +09:00 · 2024-12-25 12:50:22 +09:00 · 5fe8d92c64
parent 3043758bfd
commit 5fe8d92c64
2 changed files with 3 additions and 3 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 numpy
-torch>=1.1
-torchvision>=0.4
+torch==2.3.1
+torchvision==0.18.1
 scikit-learn
 yacs
 tqdm
--- a/setup.py
+++ b/setup.py
@ -192,7 +192,7 @@ if __name__ == '__main__':
        ],
        license='Apache License 2.0',
        install_requires=[
-            'numpy', 'torch>=1.2', 'torchvision>=0.4', 'scikit-learn', 'yacs', 'tqdm',
+            'numpy', 'torch==2.3.1', 'torchvision==0.18.1', 'scikit-learn', 'yacs', 'tqdm',
        ],
        packages=find_packages(),
        zip_safe=False)