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

model checkpoints

Browse Source
This commit is contained in:
Dobromir Popov
2025-06-26 01:12:36 +03:00
parent 0ae52f0226
commit 5e57e7817e
5 changed files with 75 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
core/orchestrator.py
View File

@ -150,12 +150,14 @@ class TradingOrchestrator:
self.model_states['dqn']['current_loss'] = checkpoint_data.get('loss', 0.0145)
self.model_states['dqn']['best_loss'] = checkpoint_data.get('best_loss', 0.0098)
self.model_states['dqn']['checkpoint_loaded'] = True
logger.info(f"DQN checkpoint loaded: loss={checkpoint_data.get('loss', 'N/A')}")
self.model_states['dqn']['checkpoint_filename'] = checkpoint_data.get('filename', 'dqn_best.pt')
logger.info(f"DQN checkpoint loaded: {checkpoint_data.get('filename', 'unknown')} loss={checkpoint_data.get('loss', 'N/A')}")
else:
# New model - set initial loss for tracking
self.model_states['dqn']['initial_loss'] = 0.285 # Typical DQN starting loss
self.model_states['dqn']['current_loss'] = 0.285
self.model_states['dqn']['best_loss'] = 0.285
self.model_states['dqn']['checkpoint_filename'] = 'none (fresh start)'
logger.info("DQN starting fresh - no checkpoint found")
logger.info(f"DQN Agent initialized: {state_size} state features, {action_size} actions")