chore: requires bagua-core 0.4-0.5 now

BaguaSys · Jul 22, 2021 · 12ea737 · 12ea737
1 parent 0c978e9
commit 12ea737
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -35,7 +35,7 @@ def check_torch_version():
         author="Kuaishou AI Platform & DS3 Lab",
         author_email="admin@mail.xrlian.com",
         install_requires=[
-            "bagua-core>=0.3,<0.4",
+            "bagua-core>=0.4,<0.5",
             "deprecation",
             "pytest-benchmark",
             "scikit-optimize",