Corrected setting callback_list

cpuguy96 · Dec 3, 2023 · 34758c0 · 34758c0
1 parent db56a03
commit 34758c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stepcovnet/executor.py b/stepcovnet/executor.py
@@ -117,7 +117,7 @@ def execute(self, input_data: inputs.TrainingInput) -> model.StepCOVNetModel:
                 input_data,
                 saved_original_weights=weights,
                 epochs=epochs_final,
-                callbacks=self.get_retraining_callbacks(hyperparameters),
+                callback_list=self.get_retraining_callbacks(hyperparameters),
             )
             self.save(
                 input_data.config, training_history=retraining_history, retrained=True