scale params

zade23 · Feb 1, 2023 · df64e86 · df64e86
1 parent c4450a3
commit df64e86
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/baselines/ray_exp/requirements.txt b/baselines/ray_exp/requirements.txt
@@ -5,4 +5,5 @@ hnswlib
 numpy
 seaborn
 pyboy
+tensorboard
 git+https://github.com/PWhiddy/mediapy.git
diff --git a/baselines/ray_exp/train_ray.py b/baselines/ray_exp/train_ray.py
@@ -30,8 +30,8 @@
     ppo.PPOConfig()
     .environment(RedGymEnv, env_config=env_config)
     .framework("torch")
-    .resources(num_gpus=1)
-    .rollouts(num_rollout_workers=3)
+    .resources(num_gpus=4)
+    .rollouts(num_rollout_workers=48)
     .training(
         model={
             "grayscale": True,
-Original file line number
+Diff line change
@@ Expand Up / @@ -5,4 +5,5 @@ hnswlib @@
     numpy
     seaborn
     pyboy
+    tensorboard
     git+https://github.com/PWhiddy/mediapy.git