From 56fcf7b234c45a7d12dafa37bc0fc94ba518a20a Mon Sep 17 00:00:00 2001 From: Carlotta Date: Mon, 9 Dec 2024 16:41:36 +0100 Subject: [PATCH] decreased batch size to allow execution in notebook --- examples/jaxsim_as_physics_engine.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/jaxsim_as_physics_engine.ipynb b/examples/jaxsim_as_physics_engine.ipynb index eb6c51246..284a935b1 100644 --- a/examples/jaxsim_as_physics_engine.ipynb +++ b/examples/jaxsim_as_physics_engine.ipynb @@ -173,7 +173,7 @@ "key = jax.random.PRNGKey(seed=0)\n", "\n", "# Initialize the simulated time.\n", - "T = jnp.arange(start=0, stop=1.0, step=model.time_step)\n", + "T = jnp.arange(start=0, stop=0.3, step=model.time_step)\n", "\n", "# Simulate\n", "for _t in T:\n", @@ -236,7 +236,7 @@ "# Then we have to create the vector of initial state\n", "\n", "# Split subkeys for sampling random initial data.\n", - "batch_size = 10\n", + "batch_size = 5\n", "key, *subkeys = jax.random.split(key=key, num=batch_size + 1)\n", "\n", "# Create the batched data.\n",