068c4068ec
* update atari.py * fix setup.py pass the pytest * fix setup.py pass the pytest * add args "render" * change the tensorboard writter * change the tensorboard writter * change device, render, tensorboard log location * change device, render, tensorboard log location * remove some wrong local files * fix some tab mistakes and the envs name in continuous/test_xx.py * add examples and point robot maze environment * fix some bugs during testing examples * add dqn network and fix some args * change back the tensorboard writter's frequency to ensure ppo and a2c can write things normally * add a warning to collector * rm some unrelated files * reformat * fix a bug in test_dqn due to the model wrong selection * change atari frame skip and observation to improve performance * readd some files * change import * modified readme * rm tensorboard log * update atari and mujoco which are ignored * rm the wrong lines
27 lines
590 B
Python
27 lines
590 B
Python
from gym.envs.registration import register
|
|
import gym
|
|
|
|
register(
|
|
id='PointMaze-v0',
|
|
entry_point='tianshou.env.mujoco.point_maze_env:PointMazeEnv',
|
|
kwargs={
|
|
"maze_size_scaling": 4,
|
|
"maze_id": "Maze2",
|
|
"maze_height": 0.5,
|
|
"manual_collision": True,
|
|
"goal": (1, 3),
|
|
}
|
|
)
|
|
|
|
register(
|
|
id='PointMaze-v1',
|
|
entry_point='tianshou.env.mujoco.point_maze_env:PointMazeEnv',
|
|
kwargs={
|
|
"maze_size_scaling": 2,
|
|
"maze_id": "Maze2",
|
|
"maze_height": 0.5,
|
|
"manual_collision": True,
|
|
"goal": (1, 3),
|
|
}
|
|
)
|