models overhaul

This commit is contained in:
Dobromir Popov
2025-07-29 19:22:04 +03:00
parent 0b5fa07498
commit b1ae557843
7 changed files with 465 additions and 430 deletions

View File

@ -605,7 +605,9 @@ class TradingOrchestrator:
action_size = self.config.rl.get("action_space", 3)
self.rl_agent = DQNAgent(
state_shape=actual_state_size, n_actions=action_size
state_shape=actual_state_size,
n_actions=action_size,
config=self.config.rl
)
self.rl_agent.to(self.device) # Move DQN agent to the determined device