From 34758c0351656c6a10360fb102368ff719397947 Mon Sep 17 00:00:00 2001 From: Chimezie Iwuanyanwu Date: Sat, 2 Dec 2023 19:48:54 -0600 Subject: [PATCH] Corrected setting callback_list --- stepcovnet/executor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stepcovnet/executor.py b/stepcovnet/executor.py index 57183c2..1095761 100644 --- 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