fix(autotune): fix bucket size switch not effective (#48)

BaguaSys · Jun 25, 2021 · 30b490a · 30b490a
1 parent 58719c7
commit 30b490a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bagua/service/autotune_service.py b/bagua/service/autotune_service.py
@@ -247,7 +247,7 @@ def register_models():
                 default_hyperparameters = BaguaHyperparameter(
                     buckets=split_bucket_by_bucket_size(
                         tensor_list,
-                        self.default_bucket_size,
+                        session["bucket_size"],
                         self.param_group_info,
                     ),
                 )