emiraran commited on
Commit
84b3ded
·
verified ·
1 Parent(s): 32d4afa

Fix motor_decoder torch.no_grad and continuous autonomous stream

Browse files
Files changed (1) hide show
  1. src/doom_agent.py +1 -1
src/doom_agent.py CHANGED
@@ -150,7 +150,7 @@ class DoomConnectomeAgent:
150
  action_binary, action_probs = self.decoder.decode(act, visual_meta, params=self.engine.params)
151
 
152
  # 4. Execute in DOOM (2 frame skip for natural, smooth, non-hyper speed)
153
- reward = self.game.make_action(action_binary, 2)
154
  self.episode_reward += reward
155
  self.total_reward += reward
156
  self.step_count += 1
 
150
  action_binary, action_probs = self.decoder.decode(act, visual_meta, params=self.engine.params)
151
 
152
  # 4. Execute in DOOM (2 frame skip for natural, smooth, non-hyper speed)
153
+ reward = self.game.make_action(action_binary, 1)
154
  self.episode_reward += reward
155
  self.total_reward += reward
156
  self.step_count += 1