Beispiel #1
0
    def __init__(self, level=None):
        super().__init__(level=level)
        self.frame_skip = 1
        self.default_input_filter = NoInputFilter()
        self.default_output_filter = NoOutputFilter()


# Roboschool
gym_roboschool_envs = [
    'inverted_pendulum', 'inverted_pendulum_swingup',
    'inverted_double_pendulum', 'reacher', 'hopper', 'walker2d',
    'half_cheetah', 'ant', 'humanoid', 'humanoid_flagrun',
    'humanoid_flagrun_harder', 'pong'
]
roboschool_v1 = {
    e: "Roboschool{}".format(lower_under_to_upper(e) + '-v1')
    for e in gym_roboschool_envs
}

# Mujoco
gym_mujoco_envs = [
    'inverted_pendulum', 'inverted_double_pendulum', 'reacher', 'hopper',
    'walker2d', 'half_cheetah', 'ant', 'swimmer', 'humanoid',
    'humanoid_standup', 'pusher', 'thrower', 'striker'
]

mujoco_v2 = {
    e: "{}".format(lower_under_to_upper(e) + '-v2')
    for e in gym_mujoco_envs
}
mujoco_v2['walker2d'] = 'Walker2d-v2'
Beispiel #2
0

# Generic parameters for vector environments such as mujoco, roboschool, bullet, etc.
class GymVectorEnvironment(GymEnvironmentParameters):
    def __init__(self, level=None):
        super().__init__(level=level)
        self.frame_skip = 1
        self.default_input_filter = NoInputFilter()
        self.default_output_filter = NoOutputFilter()


# Roboschool
gym_roboschool_envs = ['inverted_pendulum', 'inverted_pendulum_swingup', 'inverted_double_pendulum', 'reacher',
                       'hopper', 'walker2d', 'half_cheetah', 'ant', 'humanoid', 'humanoid_flagrun',
                       'humanoid_flagrun_harder', 'pong']
roboschool_v0 = {e: "{}".format(lower_under_to_upper(e) + '-v0') for e in gym_roboschool_envs}

# Mujoco
gym_mujoco_envs = ['inverted_pendulum', 'inverted_double_pendulum', 'reacher', 'hopper', 'walker2d', 'half_cheetah',
                   'ant', 'swimmer', 'humanoid', 'humanoid_standup', 'pusher', 'thrower', 'striker']

mujoco_v2 = {e: "{}".format(lower_under_to_upper(e) + '-v2') for e in gym_mujoco_envs}
mujoco_v2['walker2d'] = 'Walker2d-v2'

# Fetch
gym_fetch_envs = ['reach', 'slide', 'push', 'pick_and_place']
fetch_v1 = {e: "{}".format('Fetch' + lower_under_to_upper(e) + '-v1') for e in gym_fetch_envs}


"""
Atari Environment Components