Spaces:
Running on Zero
Running on Zero
Fix motor_decoder torch.no_grad and continuous autonomous stream
Browse files- 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,
|
| 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
|