src/aiunn/finetune/trainer.py

@@ -203,7 +203,8 @@ class aiuNNTrainer:
         """Finetune the upscaler model"""
         if self.data_loader is None:
             raise ValueError("Data not loaded. Call load_data first.")
-        
+        # setup logging
+        self._setup_logging(output_path=output_path)
         # Setup optimizer and directories
         self.optimizer = optim.Adam(self.model.parameters(), lr=lr)
         self.checkpoint_dir = os.path.join(output_path, "checkpoints")