Upgrade torch

scil-vital · Mar 18, 2024 · 7ae7027 · 7ae7027
1 parent 5328c2d
commit 7ae7027
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -20,7 +20,7 @@ requests==2.28.*
 # 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