Skip to content

Commit 2e947dd

Browse files
authored
fix ACT's bug on distributed training (#1848)
1 parent 9a00cbb commit 2e947dd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddleslim/auto_compression/compressor.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,7 @@ def __init__(self,
161161
self.default_distill_node_pair, self.model_type = self._get_model_info()
162162

163163
if self.fleet:
164-
fleet.init(is_collective=True)
164+
fleet.init()
165165

166166
if with_variable_shape(
167167
self.model_dir,

0 commit comments

Comments
 (0)