src/aiunn/finetune/trainer.py

@@ -203,8 +203,7 @@ 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")