training wip

core/extrema_trainer.py

@@ -25,7 +25,6 @@ import json
 # Import checkpoint management
 import torch
 from utils.checkpoint_manager import save_checkpoint, load_best_checkpoint
-from utils.training_integration import get_training_integration
 
 logger = logging.getLogger(__name__)
 
@@ -73,7 +72,7 @@ class ExtremaTrainer:
         # Checkpoint management
         self.model_name = model_name
         self.enable_checkpoints = enable_checkpoints
-        self.training_integration = get_training_integration() if enable_checkpoints else None
+        self.training_integration = None  # Removed dependency on utils.training_integration
         self.training_session_count = 0
         self.best_detection_accuracy = 0.0
         self.checkpoint_frequency = 50  # Save checkpoint every 50 training sessions