bugfix again for first init varaible #23

Merged
Fabel merged 1 commits from feat/bugfix into main 2025-06-30 08:43:31 +00:00
1 changed files with 3 additions and 0 deletions

View File

@ -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