Machine-Learning
/
aiuNN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request 'bugfix again for first init varaible' (#23) from feat/bugfix into main
Run VectorLoader Script / Explore-Gitea-Actions (push) Successful in 18s Details
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Successful in 9m36s Details 

Reviewed-on: #23
This commit is contained in:
Falko Victor Habel 2025-06-30 08:43:31 +00:00
parent 37c6ddc54c 8c4cd6612a
commit e01834bb67
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/aiunn/finetune/trainer.py
View File

@ -202,6 +202,9 @@ class aiuNNTrainer:
def finetune(self, output_path, epochs=10, lr=1e-4, patience=3, min_delta=0.001): def finetune(self, output_path, epochs=10, lr=1e-4, patience=3, min_delta=0.001):
"""Finetune the upscaler model""" """Finetune the upscaler model"""
is_improved = False # first the model hasn't been improved
if self.data_loader is None: if self.data_loader is None:
raise ValueError("Data not loaded. Call load_data first.") raise ValueError("Data not loaded. Call load_data first.")
# setup logging # setup logging