torch version compatible (#1162)

intelligent-machine-learning · Jun 24, 2024 · 7f05d90 · 7f05d90
1 parent c2cc6d9
commit 7f05d90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlrover/trainer/torch/utils.py b/dlrover/trainer/torch/utils.py
@@ -16,4 +16,4 @@
 
 
 def version_less_than_230():
-    return Version(torch.__version__) < Version("2.3.0")
+    return Version(torch.__version__) <= Version("2.2.2")