From 8c4cd6612a419d91ea7343b54e3f656653b7cce4 Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Mon, 30 Jun 2025 10:24:02 +0200 Subject: [PATCH] bugfix again for first init varaible --- src/aiunn/finetune/trainer.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/aiunn/finetune/trainer.py b/src/aiunn/finetune/trainer.py index a59f397..1a427bd 100644 --- a/src/aiunn/finetune/trainer.py +++ b/src/aiunn/finetune/trainer.py @@ -202,6 +202,9 @@ class aiuNNTrainer: def finetune(self, output_path, epochs=10, lr=1e-4, patience=3, min_delta=0.001): """Finetune the upscaler model""" + + is_improved = False # first the model hasn't been improved + if self.data_loader is None: raise ValueError("Data not loaded. Call load_data first.") # setup logging