popov/gogo2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

training wip

Browse Source
This commit is contained in:
Dobromir Popov
2025-09-02 19:25:13 +03:00
parent 6dcb82c184
commit 226a6aa047
8 changed files with 241 additions and 771 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
NN/models/cnn_model.py
View File

@@ -21,7 +21,6 @@ from typing import Dict, Any, Optional, Tuple
# Import checkpoint management
from utils.checkpoint_manager import save_checkpoint, load_best_checkpoint
from utils.training_integration import get_training_integration
# Configure logging
logger = logging.getLogger(__name__)
@@ -522,7 +521,7 @@ class CNNModelTrainer:
# 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.epoch_count = 0
self.best_val_accuracy = 0.0
self.best_val_loss = float('inf')

3
NN/models/dqn_agent.py
View File

@@ -16,7 +16,6 @@ sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(
# Import checkpoint management
from utils.checkpoint_manager import save_checkpoint, load_best_checkpoint
from utils.training_integration import get_training_integration
# Configure logger
logger = logging.getLogger(__name__)
@@ -44,7 +43,7 @@ class DQNAgent:
# 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.episode_count = 0
self.best_reward = float('-inf')
self.reward_history = deque(maxlen=100)