chenlin/Mujoco_WASM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Bump benchmarks from V100 to A100 to match published numbers.

Browse Source 
PiperOrigin-RevId: 576230623
Change-Id: Ib17ffd210a781e1024f8bb484c75547ee39876f7
This commit is contained in:
Erik Frey
2023-10-24 12:36:55 -07:00
committed by Copybara-Service
parent a4588cee12
commit 29f6f06788
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mjx/mujoco/mjx/benchmark/benchmark.py
+3 -3
View File
@@ -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,