From 01a9e4cdac88d5a8547b099fe804a27ca6955609 Mon Sep 17 00:00:00 2001 From: xuyxu Date: Wed, 25 Aug 2021 22:39:24 +0800 Subject: [PATCH] fix(sGBM): missing functionality of param `save_model` --- torchensemble/soft_gradient_boosting.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchensemble/soft_gradient_boosting.py b/torchensemble/soft_gradient_boosting.py index 2da5a73..ab688be 100644 --- a/torchensemble/soft_gradient_boosting.py +++ b/torchensemble/soft_gradient_boosting.py @@ -278,7 +278,7 @@ def fit( # Validation if test_loader: flag = self._evaluate_during_fit(test_loader, epoch) - if flag: + if save_model and flag: io.save(self, save_dir, self.logger) # Update the scheduler