update

PaddlePaddle · Mar 23, 2022 · c02328f · c02328f
1 parent 51ca6b2
commit c02328f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/distributed/fleet/base/fleet_base.py b/python/paddle/distributed/fleet/base/fleet_base.py
@@ -47,7 +47,7 @@
 
 class _RecomputeModelWrapper(paddle.nn.Layer):
     def __init__(self, model, segments=1, preserve_rng_state=True):
-        super(RecomputeModelWrapper, self).__init__()
+        super(_RecomputeModelWrapper, self).__init__()
         assert isinstance(model, paddle.nn.Sequential), (
             "The model passed to RecomputeModelWrapper must be of type "
             "paddle.nn.Sequential.")