• Stars
    star
    186
  • Rank 206,116 (Top 5 %)
  • Language
    Python
  • License
    MIT License
  • Created over 5 years ago
  • Updated over 1 year ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

OpenAI Gym wrapper for the DeepMind Control Suite

OpenAI Gym wrapper for the DeepMind Control Suite.

A lightweight wrapper around the DeepMind Control Suite that provides the standard OpenAI Gym interface. The wrapper allows to specify the following:

  • Reliable random seed initialization that will ensure deterministic behaviour.
  • Setting from_pixels=True converts proprioceptive observations into image-based. In additional, you can choose the image dimensions, by setting height and width.
  • Action space normalization bound each action's coordinate into the [-1, 1] range.
  • Setting frame_skip argument lets to perform action repeat.

Instalation

pip install git+git://github.com/denisyarats/dmc2gym.git

Usage

import dmc2gym

env = dmc2gym.make(domain_name='point_mass', task_name='easy', seed=1)

done = False
obs = env.reset()
while not done:
  action = env.action_space.sample()
  obs, reward, done, info = env.step(action)