[PT] bump torch to 2.5.1

README.md

@@ -439,12 +439,12 @@ conda install -c conda-forge nncf
 - Ubuntu\* 18.04 or later (64-bit)
 - Python\* 3.9 or later
 - Supported frameworks:
-  - PyTorch\* >=2.3, <2.5
+  - PyTorch\* >=2.3, <2.6
   - TensorFlow\* >=2.8.4, <=2.15.1
   - ONNX\* ==1.16.0
   - OpenVINO\* >=2022.3.0
 
-This repository is tested on Python* 3.10.14, PyTorch* 2.4.0 (NVidia CUDA\* Toolkit 12.1) and TensorFlow* 2.12.1 (NVidia CUDA\* Toolkit 11.8).
+This repository is tested on Python* 3.10.14, PyTorch* 2.5.0 (NVidia CUDA\* Toolkit 12.1) and TensorFlow* 2.12.1 (NVidia CUDA\* Toolkit 11.8).
 
 ## NNCF Compressed NNCF Model Zoo
 

constraints.txt

@@ -2,8 +2,8 @@
 openvino==2024.4.0
 
 # Pytorch
-torch==2.4.0
-torchvision==0.19.0
+torch==2.5.0
+torchvision==0.20.0
 
 # ONNX
 onnx==1.16.2

docs/Installation.md

@@ -43,7 +43,7 @@ as well as the supported versions of Python:
 
 | NNCF      | OpenVINO   | PyTorch  | ONNX     | TensorFlow | Python |
 |-----------|------------|----------|----------|------------|--------|
-| `develop` | `2024.4.0` | `2.4.0`  | `1.16.0` | `2.15.1`   | `3.10` |
+| `develop` | `2024.4.0` | `2.5.0`  | `1.16.0` | `2.15.1`   | `3.10` |
 | `2.13.0`  | `2024.4.0` | `2.4.0`  | `1.16.0` | `2.15.1`   | `3.8`* |
 | `2.12.0`  | `2024.3.0` | `2.3.0`  | `1.16.0` | `2.15.1`   | `3.8`* |
 | `2.11.0`  | `2024.2.0` | `2.3.0`  | `1.16.0` | `2.12.0`   | `3.8`  |

examples/post_training_quantization/torch/mobilenet_v2/requirements.txt

@@ -1,5 +1,5 @@
 fastdownload==0.0.7
 openvino==2024.4
 scikit-learn
-torch==2.4.0
-torchvision==0.19.0
+torch==2.5.0
+torchvision==0.20.0

examples/post_training_quantization/torch/ssd300_vgg16/requirements.txt

@@ -2,7 +2,7 @@ fastdownload==0.0.7
 onnx==1.16.0
 openvino==2024.4
 pycocotools==2.0.7
-torch==2.4.0
+torch==2.5.0
 torchmetrics==1.0.1
-torchvision==0.19.0
+torchvision==0.20.0
 numpy<2

examples/quantization_aware_training/torch/resnet18/requirements.txt

@@ -1,4 +1,4 @@
 fastdownload==0.0.7
 openvino==2024.4
-torch==2.4.0
-torchvision==0.19.0
+torch==2.5.0
+torchvision==0.20.0

nncf/experimental/torch2/function_hook/hook_executor_mode.py

@@ -106,7 +106,7 @@ def __init__(self, model: nn.Module, hook_storage: HookStorage) -> None:
         :param model: The PyTorch model to which the hooks will be applied.
         :param hook_storage: Storage for hooks to be executed.
         """
-        super().__init__()  # type: ignore
+        super().__init__()
         self.hook_storage: HookStorage = hook_storage
         self.model: nn.Module = model
         self.module_call_stack: List[nn.Module] = []

nncf/version.py

@@ -12,6 +12,6 @@
 __version__ = "2.14.0"
 
 
-BKC_TORCH_SPEC = "==2.4.*"
+BKC_TORCH_SPEC = "==2.5.*"
 BKC_TF_SPEC = "==2.15.*"
 STRICT_TF_SPEC = ">=2.9.3,<2.16.0"