bug fix

PaddlePaddle · Nov 29, 2023 · 90e68a6 · 90e68a6
1 parent b779c10
commit 90e68a6
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/python/paddle/distributed/__init__.py b/python/paddle/distributed/__init__.py
@@ -84,7 +84,6 @@
     dtensor_from_fn,
     reshard,
     shard_layer,
-    shard_optimizer,
     ShardOptimizer,
 )
 
@@ -159,6 +158,5 @@
     "Shard",
     "Replicate",
     "Partial",
-    "shard_optimizer",
     "ShardOptimizer",
 ]