From a24ad6390573a55a0af938cc751f6ce42d86b4df Mon Sep 17 00:00:00 2001 From: Lev Kozlov Date: Tue, 17 Sep 2024 19:02:32 +0900 Subject: [PATCH] fix: increase batch --- benchmarks/run_benchmark.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/benchmarks/run_benchmark.py b/benchmarks/run_benchmark.py index 2e191db..4f09479 100644 --- a/benchmarks/run_benchmark.py +++ b/benchmarks/run_benchmark.py @@ -43,11 +43,11 @@ class ColabPaths: torch.manual_seed(0) -N_ENVS_SWEEP = [1, 2, 4, 8, 16, 32, 64, 128, 256] # , 512, 1024, 2048, 4096, 8192, 16384, 32768] +N_ENVS_SWEEP = [2**i for i in range(20)] N_EVALS = 20 # Load functions for CUDA benchmarking -fn_files = ["fn_1e1.casadi", "fn_1e2.casadi", "fn_1e3.casadi"] # , "fn_1e4.casadi", "fn_1e5.casadi"] +fn_files = ["fn_1e1.casadi", "fn_1e2.casadi"] benchmark_fns = [Function.load(os.path.join(PathsProvider.FUNCTIONS_DIR, fn)) for fn in fn_files]