Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit a12519b

Browse files
authoredFeb 23, 2024··
Merge branch 'develop' into yolo_box-lianghua
2 parents 167dbdc + 2e947dd commit a12519b

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)
Please sign in to comment.