Upgrade torch

EmmaRenauld · Apr 25, 2024 · 2509a8c · 2509a8c
1 parent d7711a1
commit 2509a8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -14,7 +14,7 @@ scilpy==2.0
 # Other important dependencies
 # -------
 bertviz==1.4.0  # For transformer's visu
-torch==1.13.*
+torch==2.2.0
 tqdm==4.64.*
 comet-ml>=3.22.0
 contextlib2==21.6.0