dodox/Columbus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Made Value-Func Rendering way more efficient

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-10-17 23:00:01 +02:00
parent ba796a1d6e
commit 89e5f1369e
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
columbus/env.py
View File

@ -316,7 +316,7 @@ class ColumbusEnv(gym.Env):
self.agent.pos = agentpos self.agent.pos = agentpos
self.agent.speed = agentspeed self.agent.speed = agentspeed
V = value_func(th.Tensor(obs)) V = value_func(th.Tensor(np.array(obs)))
V /= max(V.max(), -1*V.min())*2 V /= max(V.max(), -1*V.min())*2
V += 0.5 V += 0.5
@ -408,6 +408,8 @@ class ColumbusEnv(gym.Env):
self.draw_confidence_ellipse = not self.draw_confidence_ellipse self.draw_confidence_ellipse = not self.draw_confidence_ellipse
elif keys[pygame.K_r]: elif keys[pygame.K_r]:
self.reset() self.reset()
elif keys[pygame.K_t]:
self._reset_paths()
elif keys[pygame.K_p]: elif keys[pygame.K_p]:
self.paused = not self.paused self.paused = not self.paused
else: else: