Updated some stuff

2023-11-20 01:51:54 +01:00 · 2023-11-20 01:51:54 +01:00 · 3fb147afff
commit 3fb147afff
parent da649ccca8
3 changed files with 1 additions and 2 deletions
--- a/agents/ppo/agent.py
+++ b/agents/ppo/agent.py
@ -1,4 +1,3 @@
-import os
 import numpy as np
 import torch as T

@ -40,7 +39,7 @@ class Agent:
        print('.. done ...')

    def choose_action(self, observation):
-        state = T.tensor([observation], dtype=T.float).to(self.actor.device)
+        state = T.tensor(observation, dtype=T.float).to(self.actor.device)

        dist = self.actor(state)
        value = self.critic(state)
--- a/tmp/ppo/actor_torch_ppo
+++ b/tmp/ppo/actor_torch_ppo
--- a/tmp/ppo/critic_torch_ppo
+++ b/tmp/ppo/critic_torch_ppo