def train_config(self): disable_view_window() c = Config() c.env_name = "Pendulum-v0" c.env = unwrap_time_limit(gym.make(c.env_name)) c.observe_dim = 3 c.action_dim = 1 c.action_range = 2 c.max_episodes = 1000 c.max_steps = 200 c.replay_size = 100000 c.solved_reward = -400 c.solved_repeat = 5 return c
def train_config(self, pytestconfig): disable_view_window() c = Config() c.env_name = "Pendulum-v0" c.env = unwrap_time_limit(gym.make(c.env_name)) c.observe_dim = 3 c.action_dim = 1 c.action_range = 2 c.max_episodes = 1000 c.max_steps = 200 c.noise_param = (0, 0.2) c.noise_mode = "normal" c.noise_interval = 2 c.replay_size = 100000 c.solved_reward = -150 c.solved_repeat = 5 c.device = "cpu" return c