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

env v0.11

Browse Source
This commit is contained in:
张德祥 2023-06-13 10:44:54 +08:00
parent 7879c6cfe7
commit 5038a91aad
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dreamer.py
View File

@ -213,7 +213,7 @@ def make_env(config, logger, mode, train_eps, eval_eps):
elif suite == "mazegym": elif suite == "mazegym":
import gym import gym
env = gym.make('memory_maze:MemoryMaze-9x9-v0') env = gym.make('memory_maze:MemoryMaze-9x9-v0')
from envs.memmaze import MZGymWrapper from envs.memmazeEnv import MZGymWrapper
env = MZGymWrapper(env) env = MZGymWrapper(env)
env = wrappers.OneHotAction(env) env = wrappers.OneHotAction(env)

2
envs/memmaze.py → envs/memmazeEnv.py
View File

@ -9,7 +9,7 @@ import gym
import numpy as np import numpy as np
class GymWrapper: class MZGymWrapper:
def __init__(self, env, obs_key='image', act_key='action'): def __init__(self, env, obs_key='image', act_key='action'):
self._env = env self._env = env