From 29f6f06788651d7d6262573edca13ae2f83b04a7 Mon Sep 17 00:00:00 2001 From: Erik Frey Date: Tue, 24 Oct 2023 12:36:55 -0700 Subject: [PATCH] Bump benchmarks from V100 to A100 to match published numbers. PiperOrigin-RevId: 576230623 Change-Id: Ib17ffd210a781e1024f8bb484c75547ee39876f7 --- mjx/mujoco/mjx/benchmark/benchmark.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mjx/mujoco/mjx/benchmark/benchmark.py b/mjx/mujoco/mjx/benchmark/benchmark.py index b0a59af0..61112953 100644 --- a/mjx/mujoco/mjx/benchmark/benchmark.py +++ b/mjx/mujoco/mjx/benchmark/benchmark.py @@ -38,9 +38,9 @@ _BATCH_SIZE = { ('humanoid', 'TPU v5 lite'): 1024, ('barkour', 'TPU v5 lite'): 1024, ('shadow_hand', 'TPU v5 lite'): 1024, - ('humanoid', 'Tesla V100-SXM2-16GB'): 8192, - ('barkour', 'Tesla V100-SXM2-16GB'): 8192, - ('shadow_hand', 'Tesla V100-SXM2-16GB'): 4096, + ('humanoid', 'NVIDIA A100-SXM4-40GB'): 8192, + ('barkour', 'NVIDIA A100-SXM4-40GB'): 8192, + ('shadow_hand', 'NVIDIA A100-SXM4-40GB'): 4096, ('humanoid', 'cpu'): 64, ('barkour', 'cpu'): 64, ('shadow_hand', 'cpu'): 64,