hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

policy is not given logs

Browse Source
This commit is contained in:
NM512 2023-10-01 06:25:23 +09:00
parent 37da2d3498
commit f35480f2a6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools.py
View File

@ -164,7 +164,7 @@ def simulate(
# replace obs with done by initial state # replace obs with done by initial state
obs[index] = result obs[index] = result
# step agents # step agents
obs = {k: np.stack([o[k] for o in obs]) for k in obs[0]} obs = {k: np.stack([o[k] for o in obs]) for k in obs[0] if "log_" not in k}
action, agent_state = agent(obs, done, agent_state) action, agent_state = agent(obs, done, agent_state)
if isinstance(action, dict): if isinstance(action, dict):
action = [ action = [