From 265b84c45bd165e74dfbace0be92a77478e7ad2a Mon Sep 17 00:00:00 2001
From: Lev Kozlov <kozlov.l.a10@gmail.com>
Date: Tue, 17 Sep 2024 19:25:03 +0900
Subject: [PATCH] fix: reduce

---
 benchmarks/run_benchmark.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/benchmarks/run_benchmark.py b/benchmarks/run_benchmark.py
index 9d8edc8..30a6b57 100644
--- 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