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

updated the logging process
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Successful in 42s Details

This commit is contained in:
Falko Victor Habel 2025-04-23 14:29:18 +02:00
parent f3e59a6586
commit 3b140d559c
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")