torch version compatible

intelligent-machine-learning · Jun 19, 2024 · 5872724 · 5872724
1 parent e0e095f
commit 5872724
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")