diff --git a/inst/python/Estimator.py b/inst/python/Estimator.py index d9a92ee..69ee153 100644 --- a/inst/python/Estimator.py +++ b/inst/python/Estimator.py @@ -99,7 +99,7 @@ def __init__(self, model, model_parameters, estimator_settings): self.best_score = None self.best_epoch = None self.learn_rate_schedule = None - if parameters["estimator_settings"]["compile"]: + if estimator_settings["compile"]: self.model = torch.compile(self.model, dynamic=False) def fit(self, dataset, test_dataset):