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

mem maze env ok 1.1

Browse Source
This commit is contained in:
zdx 2023-06-17 23:59:05 +08:00
parent ea446adaf4
commit 152415f32e
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs.yaml
View File

@ -157,7 +157,7 @@ debug:
batch_length: 20 batch_length: 20
mazegym: mazegym:
task: 9 task: '9'
steps: 1e6 steps: 1e6
action_repeat: 2 action_repeat: 2

6
dreamer.py
View File

@ -212,17 +212,15 @@ def make_env(config, logger, mode, train_eps, eval_eps):
env = wrappers.OneHotAction(env) env = wrappers.OneHotAction(env)
elif suite == "mazegym": elif suite == "mazegym":
import gym import gym
if task == 9: if task == '9':
env = gym.make('memory_maze:MemoryMaze-9x9-v0') env = gym.make('memory_maze:MemoryMaze-9x9-v0')
elif task == 15: elif task == '15':
env = gym.make('memory_maze:MemoryMaze-15x15-v0') env = gym.make('memory_maze:MemoryMaze-15x15-v0')
else: else:
raise NotImplementedError(suite) raise NotImplementedError(suite)
from envs.memmazeEnv import MZGymWrapper from envs.memmazeEnv import MZGymWrapper
env = MZGymWrapper(env) env = MZGymWrapper(env)
#from envs.memmazeEnv import OneHotAction as OneHotAction2
env = wrappers.OneHotAction2(env) env = wrappers.OneHotAction2(env)
else: else:
raise NotImplementedError(suite) raise NotImplementedError(suite)
env = wrappers.TimeLimit(env, config.time_limit) env = wrappers.TimeLimit(env, config.time_limit)