fix: dont instantiate if estimators loaded from save_dir

TorchEnsemble-Community · Jun 14, 2024 · 460c382 · 460c382
1 parent 6726a99
commit 460c382
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/torchensemble/soft_gradient_boosting.py b/torchensemble/soft_gradient_boosting.py
@@ -212,8 +212,10 @@ def fit(
     ):
 
         # Instantiate base estimators and set attributes
-        for _ in range(self.n_estimators):
-            self.estimators_.append(self._make_estimator())
+        # dont instantiate if estimators loaded from save_dir
+        if len(self.estimators_) != self.n_estimators:
+            for _ in range(self.n_estimators):
+                self.estimators_.append(self._make_estimator())
         self._validate_parameters(epochs, log_interval)
         self.n_outputs = self._decide_n_outputs(train_loader)