refactoring
This commit is contained in:
@@ -35,7 +35,7 @@ logging.basicConfig(
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# Import checkpoint management
|
||||
from utils.checkpoint_manager import get_checkpoint_manager, get_checkpoint_stats
|
||||
from NN.training.model_manager import create_model_manager
|
||||
from utils.training_integration import get_training_integration
|
||||
|
||||
# Import training components
|
||||
@@ -55,7 +55,7 @@ class CheckpointIntegratedTrainingSystem:
|
||||
self.running = False
|
||||
|
||||
# Checkpoint management
|
||||
self.checkpoint_manager = get_checkpoint_manager()
|
||||
self.checkpoint_manager = create_model_manager()
|
||||
self.training_integration = get_training_integration()
|
||||
|
||||
# Data provider
|
||||
|
||||
Reference in New Issue
Block a user