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

Merge pull request 'updated the logging process' (#16) from feat/checkpoints into main
Run VectorLoader Script / Explore-Gitea-Actions (push) Successful in 20s Details
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Successful in 38s Details 

Reviewed-on: #16
This commit is contained in:
Falko Victor Habel 2025-04-23 12:32:46 +00:00
parent fd924fa024 3b140d559c
commit 9b82024a40
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -203,7 +203,8 @@ class aiuNNTrainer:
"""Finetune the upscaler model""" """Finetune the upscaler model"""
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
self._setup_logging(output_path=output_path)
# Setup optimizer and directories # Setup optimizer and directories
self.optimizer = optim.Adam(self.model.parameters(), lr=lr) self.optimizer = optim.Adam(self.model.parameters(), lr=lr)
self.checkpoint_dir = os.path.join(output_path, "checkpoints") self.checkpoint_dir = os.path.join(output_path, "checkpoints")