sharding optimizer color edit form int to dict; test=develop

PaddlePaddle · Jan 10, 2025 · 3f400bd · 3f400bd
1 parent 948b7e2
commit 3f400bd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test/collective/fleet/hybrid_parallel_sharding_model.py b/test/collective/fleet/hybrid_parallel_sharding_model.py
@@ -183,7 +183,7 @@ def __init__(
                 p.color = {'color': "linear1"}
 
             for p in self.linear2.parameters():
-                p.color = {'color': "linear1"}
+                p.color = {'color': "linear2"}
 
             for p in self.linear3.parameters():
                 p.color = {'color': "linear3"}

diff --git a/test/collective/fleet/hybrid_parallel_sharding_model_with_fuse_optimizer_states_enabled.py b/test/collective/fleet/hybrid_parallel_sharding_model_with_fuse_optimizer_states_enabled.py
@@ -126,7 +126,7 @@ def __init__(
                 p.color = {'color': "linear1"}
 
             for p in self.linear2.parameters():
-                p.color = {'color': "linear1"}
+                p.color = {'color': "linear2"}
 
             for p in self.linear3.parameters():
                 p.color = {'color': "linear3"}