fix: reduce

based-robotics · Sep 17, 2024 · 265b84c · 265b84c
1 parent c84d1f3
commit 265b84c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/run_benchmark.py b/benchmarks/run_benchmark.py
@@ -43,7 +43,7 @@ class ColabPaths:
 
     torch.manual_seed(0)
 
-N_ENVS_SWEEP = [2**i for i in range(50)]
+N_ENVS_SWEEP = [2**i for i in range(21)]
 N_EVALS = 20
 
 # Load functions for CUDA benchmarking