Umaze-Diverse

Description

The data is collected from the AntMaze_UMaze-v4 environment, which contains a U shape maze. At the beginning of each episode random locations for the goal and agent’s reset are selected. The success rate of all the trajectories is more than 90%, failed trajectories occur because the Ant flips and can’t stand up again. As in v1 the reward is sparse, but the data is collected with continuing_task=False: the episode terminates as soon as the Ant reaches the goal, so the reward can be collected only once and the maximum return of an episode is 1. The Ant reaches the goals by following a set of waypoints using a goal-reaching policy trained using SAC.

Dataset Specs

Total Steps	1000000
Total Episodes	7223
Dataset Observation Space	Dict('achieved_goal': Box(-inf, inf, (2,), float64), 'desired_goal': Box(-inf, inf, (2,), float64), 'observation': Box(-inf, inf, (27,), float64))
Dataset Action Space	Box(-1.0, 1.0, (8,), float32)
Algorithm	QIteration+SAC
Author	Alex Davey
Email	alexdavey0@gmail.com
Code Permalink	https://github.com/rodrigodelazcano/d4rl-minari-dataset-generation
Minari Version	0.4.3 (supported)
Download	minari download D4RL/antmaze/umaze-diverse-v2

Environment Specs

The following table rows correspond to the Gymnasium environment specifications used to generate the dataset. To read more about what each parameter means you can have a look at the Gymnasium documentation https://gymnasium.farama.org/api/registry/#gymnasium.envs.registration.EnvSpec

This environment can be recovered from the Minari dataset as follows:

import minari

dataset = minari.load_dataset('D4RL/antmaze/umaze-diverse-v2')
env  = dataset.recover_environment()

ID	AntMaze_UMaze-v4
Observation Space	Dict('achieved_goal': Box(-inf, inf, (2,), float64), 'desired_goal': Box(-inf, inf, (2,), float64), 'observation': Box(-inf, inf, (27,), float64))
Action Space	Box(-1.0, 1.0, (8,), float32)
entry_point	gymnasium_robotics.envs.maze.ant_maze_v4:AntMazeEnv
max_episode_steps	700
reward_threshold	None
nondeterministic	False
order_enforce	True
disable_env_checker	False
kwargs	{'maze_map': [[1, 1, 1, 1, 1], [1, 0, 0, 0, 1], [1, 1, 1, 0, 1], [1, 0, 0, 0, 1], [1, 1, 1, 1, 1]], 'reward_type': 'sparse', 'continuing_task': False}
additional_wrappers	()
vector_entry_point	None

Evaluation Environment Specs

This environment can be recovered from the Minari dataset as follows:

import minari

dataset = minari.load_dataset('D4RL/antmaze/umaze-diverse-v2')
eval_env  = dataset.recover_environment(eval_env=True)

ID	AntMaze_UMaze-v4
Observation Space	Dict('achieved_goal': Box(-inf, inf, (2,), float64), 'desired_goal': Box(-inf, inf, (2,), float64), 'observation': Box(-inf, inf, (27,), float64))
Action Space	Box(-1.0, 1.0, (8,), float32)
entry_point	gymnasium_robotics.envs.maze.ant_maze_v4:AntMazeEnv
max_episode_steps	700
reward_threshold	None
nondeterministic	False
order_enforce	True
disable_env_checker	False
kwargs	{'maze_map': [[1, 1, 1, 1, 1], [1, 0, 0, 'r', 1], [1, 0, 1, 1, 1], [1, 0, 0, 'g', 1], [1, 1, 1, 1, 1]], 'reward_type': 'sparse', 'continuing_task': False}
additional_wrappers	()
vector_entry_point	None