models overhaul
This commit is contained in:
@ -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
|
||||
|
||||
|
Reference in New Issue
Block a user