mirror of
https://github.com/wassname/ray.git
synced 2026-07-13 08:10:21 +08:00
[rllib] Move repeat field to asv script (#2367)
This commit is contained in:
@@ -52,7 +52,6 @@ class LocalMultiGPUOptimizer(PolicyOptimizer):
|
||||
self.update_weights_timer = TimerStat()
|
||||
|
||||
print("LocalMultiGPUOptimizer devices", self.devices)
|
||||
print("LocalMultiGPUOptimizer batch size", self.batch_size)
|
||||
|
||||
assert set(self.local_evaluator.policy_map.keys()) == {"default"}, \
|
||||
"Multi-agent is not supported"
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
cartpole-a3c:
|
||||
env: CartPole-v0
|
||||
run: A3C
|
||||
repeat: 3
|
||||
stop:
|
||||
episode_reward_mean: 200
|
||||
time_total_s: 600
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
cartpole-a3c:
|
||||
env: CartPole-v0
|
||||
run: A3C
|
||||
repeat: 3
|
||||
stop:
|
||||
episode_reward_mean: 200
|
||||
time_total_s: 600
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
cartpole-dqn:
|
||||
env: CartPole-v0
|
||||
run: DQN
|
||||
repeat: 3
|
||||
stop:
|
||||
episode_reward_mean: 200
|
||||
time_total_s: 600
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
cartpole-pg:
|
||||
env: CartPole-v0
|
||||
run: PG
|
||||
repeat: 3
|
||||
stop:
|
||||
episode_reward_mean: 200
|
||||
time_total_s: 300
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
cartpole-ppo:
|
||||
env: CartPole-v0
|
||||
run: PPO
|
||||
repeat: 3
|
||||
stop:
|
||||
episode_reward_mean: 200
|
||||
time_total_s: 300
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
pendulum-ddpg:
|
||||
env: Pendulum-v0
|
||||
run: DDPG
|
||||
repeat: 3
|
||||
stop:
|
||||
episode_reward_mean: -160
|
||||
time_total_s: 900
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
pendulum-ppo:
|
||||
env: Pendulum-v0
|
||||
run: PPO
|
||||
repeat: 3
|
||||
stop:
|
||||
episode_reward_mean: -160
|
||||
# expect -140 within 300-500k steps
|
||||
|
||||
@@ -22,6 +22,8 @@ CONFIG_DIR = os.path.dirname(os.path.abspath(__file__))
|
||||
def _evaulate_config(filename):
|
||||
with open(os.path.join(CONFIG_DIR, filename)) as f:
|
||||
experiments = yaml.load(f)
|
||||
for _, config in experiments.items():
|
||||
config["repeat"] = 3
|
||||
ray.init()
|
||||
trials = tune.run_experiments(experiments)
|
||||
results = defaultdict(list)
|
||||
|
||||
Reference in New Issue
Block a user