fix(regression): fix scheduler

dreamquark-ai · Dec 6, 2019 · 01e46b7 · 01e46b7
1 parent 91461fb
commit 01e46b7
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/pytorch_tabnet/tab_model.py b/pytorch_tabnet/tab_model.py
@@ -698,11 +698,6 @@ def update_fit_params(self, X_train, y_train, X_valid, y_valid, loss_fn,
  self.weights = 0 # No weights for regression
  self.updated_weights = 0
 
- if self.scheduler_fn:
- self.scheduler = self.scheduler_fn(self.optimizer, **self.scheduler_params)
- else:
- self.scheduler = None
-
  self.max_epochs = max_epochs
  self.patience = patience
  self.batch_size = batch_size