[misc] fix ds config (#7205)

Former-commit-id: db113f690e
2025-12-14 19:06:26 +08:00 · 2025-03-07 15:21:28 +08:00
parent 33b4c33279
commit 98ea0e8109
6 changed files with 6 additions and 6 deletions
--- a/examples/deepspeed/ds_z0_config.json
+++ b/examples/deepspeed/ds_z0_config.json
@@ -19,7 +19,7 @@
    "stage": 0,
    "allgather_partitions": true,
    "allgather_bucket_size": 5e8,
-    "overlap_comm": true,
+    "overlap_comm": false,
    "reduce_scatter": true,
    "reduce_bucket_size": 5e8,
    "contiguous_gradients": true,
--- a/examples/deepspeed/ds_z2_config.json
+++ b/examples/deepspeed/ds_z2_config.json
@@ -19,7 +19,7 @@
    "stage": 2,
    "allgather_partitions": true,
    "allgather_bucket_size": 5e8,
-    "overlap_comm": true,
+    "overlap_comm": false,
    "reduce_scatter": true,
    "reduce_bucket_size": 5e8,
    "contiguous_gradients": true,
--- a/examples/deepspeed/ds_z2_offload_config.json
+++ b/examples/deepspeed/ds_z2_offload_config.json
@@ -23,7 +23,7 @@
    },
    "allgather_partitions": true,
    "allgather_bucket_size": 5e8,
-    "overlap_comm": true,
+    "overlap_comm": false,
    "reduce_scatter": true,
    "reduce_bucket_size": 5e8,
    "contiguous_gradients": true,
--- a/examples/deepspeed/ds_z3_config.json
+++ b/examples/deepspeed/ds_z3_config.json
@@ -17,7 +17,7 @@
  },
  "zero_optimization": {
    "stage": 3,
-    "overlap_comm": true,
+    "overlap_comm": false,
    "contiguous_gradients": true,
    "sub_group_size": 1e9,
    "reduce_bucket_size": "auto",
--- a/examples/deepspeed/ds_z3_offload_config.json
+++ b/examples/deepspeed/ds_z3_offload_config.json
@@ -25,7 +25,7 @@
      "device": "cpu",
      "pin_memory": true
    },
-    "overlap_comm": true,
+    "overlap_comm": false,
    "contiguous_gradients": true,
    "sub_group_size": 1e9,
    "reduce_bucket_size": "auto",