Bump Python requirement torch to 2.1.0

33f79fc9 · Teklia Bot · Yoann Schneider · c366007a · 33f79fc9 · 33f79fc9
Commit 33f79fc9 authored 1 year ago by Teklia Bot Committed by Yoann Schneider 1 year ago
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,4 @@ opencv-python-headless==4.7.0.72
 pyyaml==6.0.1
 requests>=2,<3
 teklia_toolbox==0.1.3
-torch==2.0.0
+torch==2.1.0
--- a/training-requirements.txt
+++ b/training-requirements.txt
@@ -4,5 +4,5 @@ mlflow==2.2.1
 Shapely==2.0.2
 teklia-toolbox==0.1.3
 tensorboard==2.12.1
-torchvision==0.15.1
+torchvision==0.16.0
 tqdm==4.66.1