update config (#24)

* update config

* remove config log

Author: budzianowski

train.py

@@ -302,6 +302,20 @@ def __init__(
 class HumanoidWalkingTaskConfig(ksim.PPOConfig):
     """Config for the humanoid walking task."""
 
+    # Task parameters.
+    num_envs: int = xax.field(
+        value=1,
+        help="The number of environments to run in parallel.",
+    )
+    batch_size: int = xax.field(
+        value=1,
+        help="The batch size for the PPO training.",
+    )
+
+    rollout_length_seconds: float = xax.field(
+        value=1.0,
+        help="The length of the rollout in seconds.",
+    )
     # Model parameters.
     hidden_size: int = xax.field(
         value=128,
@@ -621,7 +635,6 @@ def sample_action(
             commands=commands,
             carry=actor_carry_in,
         )
-
         action_j = action_dist_j.mode() if argmax else action_dist_j.sample(seed=rng)
 
         return ksim.Action(