def __init__(self, random=None):
self.distance_threshold = 1.0
self.same_color_radius = 0.5
self.targets = np.array([[3.4, 3.4], [-3.4, 3.4]])
self.correct_idx = 0
self.goal_color_center = self.np_random.uniform(-1.0, 1.0, size=3)
self.goal_specific_color = self._sample_color_within_radius(
self.goal_color_center, self.same_color_radius)
other_center = self._sample_color_with_min_dist(
self.goal_color_center, self.same_color_radius)
self.other_specific_color = other_center
MetaMujocoEnv.__init__(self, 'low_gear_ratio_ant.xml', 5)
utils.EzPickle.__init__(self)
def __init__(self, use_relative_pos=False):
self.ACCEPT_RADIUS = 0.5
self.true_label = 0
self.task_hyperplane = np.ones(4) / 2.0
self.first_sample = np.random.uniform(size=4)
self.second_sample = np.random.uniform(size=4)
self.targets = np.array(
[
[1.41, 1.41],
[-1.41, 1.41]
]
)
self.use_relative_pos = use_relative_pos
MetaMujocoEnv.__init__(self, 'low_gear_ratio_ant.xml', 5)
utils.EzPickle.__init__(self)
def __init__(self):
self.multipliers = {
'body_mass_multiplier': np.ones(5),
'body_inertia_multiplier': np.ones((5, 3)),
'dof_damping_multiplier': np.ones(6),
'geom_friction_multiplier': np.ones((5, 3))
}
MetaMujocoEnv.__init__(self, 'hopper.xml', 4)
utils.EzPickle.__init__(self)
self.original_params = {
'body_mass': self.model.body_mass.copy(),
'body_inertia': self.model.body_inertia.copy(),
'dof_damping': self.model.dof_damping.copy(),
'geom_friction': self.model.geom_friction.copy()
}
def __init__(self):
self.multipliers = {
'body_mass_multiplier': np.ones(8),
'body_inertia_multiplier': np.ones((8, 3)),
'dof_damping_multiplier': np.ones(9),
'geom_friction_multiplier': np.ones((8, 3))
}
# MetaMujocoEnv.__init__(self, 'walker2d.xml', 5)
MetaMujocoEnv.__init__(self, 'walker_2d_with_camera.xml', 5)
utils.EzPickle.__init__(self)
self.original_params = {
'body_mass': self.model.body_mass.copy(),
'body_inertia': self.model.body_inertia.copy(),
'dof_damping': self.model.dof_damping.copy(),
'geom_friction': self.model.geom_friction.copy()
}
def __init__(self):
self.goal_direction = np.array([1.0, 0.0])
# MetaMujocoEnv.__init__(self, 'ant.xml', 5)
MetaMujocoEnv.__init__(self, 'low_gear_ratio_ant.xml', 5)
# MetaMujocoEnv.__init__(self, 'low_gear_ratio_ant.xml', 5)
utils.EzPickle.__init__(self)
def __init__(self):
self.goal_pos = np.array([14.1, 14.1])
self.target_distance = 25.0
MetaMujocoEnv.__init__(self, 'low_gear_ratio_ant_8_star.xml', 5)
utils.EzPickle.__init__(self)
def __init__(self):
self.goal_pos = np.array([1.41, 1.41])
# MetaMujocoEnv.__init__(self, 'ant.xml', 5)
MetaMujocoEnv.__init__(self, 'low_gear_ratio_ant.xml', 5)
utils.EzPickle.__init__(self)