Setting torch/torchvision versions to avoid FutureWarning, ref. ultra…

…lytics/yolov5#13244
wombat70 · Sep 16, 2024 · 8f9bf0c · 8f9bf0c
1 parent 57a33bd
commit 8f9bf0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -6,8 +6,8 @@ Pillow>=10.4.0
 PyYAML>=6.0.2
 requests>=2.32.3
 scipy>=1.14.1
-torch>=2.4.1
-torchvision>=0.19.1
+torch=2.3.1
+torchvision=0.18.1
 tqdm>=4.66.5
 #protobuf<4.21.3  # https://github.com/ultralytics/yolov5/issues/8012
 pandas>=2.2.2