Pneuma/pneuma-pygame
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Updated some stuff

Browse source
This commit is contained in:
Vasilis Valatsos 2023-11-20 01:51:54 +01:00
parent da649ccca8
commit 3fb147afff
3 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
agents/ppo/agent.py
View file

@ -1,4 +1,3 @@
import os
import numpy as np import numpy as np
import torch as T import torch as T
@ -40,7 +39,7 @@ class Agent:
print('.. done ...') print('.. done ...')
def choose_action(self, observation): 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) dist = self.actor(state)
value = self.critic(state) value = self.critic(state)

BIN
tmp/ppo/actor_torch_ppo
View file

Binary file not shown.

BIN
tmp/ppo/critic_torch_ppo
View file

Binary file not shown.