sensor

oist · Dec 15, 2023 · 672442e · 672442e
1 parent d581cf7
commit 672442e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/smoke-tests/circle_ppo.py b/smoke-tests/circle_ppo.py
@@ -230,6 +230,8 @@ def train(
     minibatch_size: int = 128,
     n_rollout_steps: int = 1024,
     n_total_steps: int = 1024 * 1000,
+    n_sensors: int = 16,
+    sensor_length: float = 100.0,
     food_loc_fn: str = "gaussian",
     env_shape: str = "circle",
     reset_interval: Optional[int] = None,
@@ -259,6 +261,8 @@ def train(
         angular_damping=angular_damping,
         max_force=max_force,
         min_force=min_force,
+        n_agent_sensors=n_sensors,
+        sensor_length=sensor_length,
     )
     network = run_training(
         jax.random.PRNGKey(seed),