remove dygraph check

PaddlePaddle · Nov 30, 2023 · 8e68443 · 8e68443
1 parent 7ab6fb9
commit 8e68443
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/python/paddle/distributed/auto_parallel/api.py b/python/paddle/distributed/auto_parallel/api.py
@@ -410,9 +410,6 @@ def replicate_layer_params_and_buffers(
 
 class _ShardOptimizer:
     def __init__(self, optimizer, shard_fn=None):
-        assert (
-            paddle.in_dynamic_mode()
-        ), "`paddle.distributed.ShardOptimizer` only supports dynamic mode for now."
         assert (
             optimizer is not None
         ), "The argument `optimizer` cannot be empty."