fix ACT's bug on distributed training (PaddlePaddle#1848)

lizexu123 · Feb 22, 2024 · 2e947dd · 2e947dd
1 parent 9a00cbb
commit 2e947dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddleslim/auto_compression/compressor.py b/paddleslim/auto_compression/compressor.py
@@ -161,7 +161,7 @@ def __init__(self,
         self.default_distill_node_pair, self.model_type = self._get_model_info()
 
         if self.fleet:
-            fleet.init(is_collective=True)
+            fleet.init()
 
         if with_variable_shape(
                 self.model_dir,