Robotics
PyTorch
world-action-model
libero

Model card: link the ROCm blog, label success rate

#2
by Viani - opened
Files changed (2) hide show
  1. ASSET_SHA256SUMS +1 -1
  2. README.md +4 -4
ASSET_SHA256SUMS CHANGED
@@ -10,5 +10,5 @@ d2023cc6a5a33abd405c3a9f63b7fabc84df6c917798bd7080114bcbe9294d89 stats/libero_s
10
  905ccc23cf99504e6f17f9fac2b5115896304be2d2dc0a9522f8886095c0548c videos/libero-object.mp4
11
  b5393c1ae185b1d05ad79edde9b5dc7b10543694b5c6be0d44084362e962e51d videos/libero-goal.mp4
12
  38e0b77c8ec5988f3357ad0e28b5456b42c95b5698e96961d5d0cb2f9433751a videos/libero-long.mp4
13
- cf481fdf2c4e2dc6ceda595b933ca210c98465cc743bcff7155d434bb459b10d README.md
14
  f8337593ad4c0174982bcd6219ba666128bf5b2bf587c338921b89e2dfd7d1ff replay.mp4
 
10
  905ccc23cf99504e6f17f9fac2b5115896304be2d2dc0a9522f8886095c0548c videos/libero-object.mp4
11
  b5393c1ae185b1d05ad79edde9b5dc7b10543694b5c6be0d44084362e962e51d videos/libero-goal.mp4
12
  38e0b77c8ec5988f3357ad0e28b5456b42c95b5698e96961d5d0cb2f9433751a videos/libero-long.mp4
13
+ 3d0847cc2c41e20eb251097bc6325b93f9d009e14bd63a4f5a36b899b9f52a77 README.md
14
  f8337593ad4c0174982bcd6219ba666128bf5b2bf587c338921b89e2dfd7d1ff replay.mp4
README.md CHANGED
@@ -16,7 +16,7 @@ pipeline_tag: robotics
16
  A latent world-action policy for the [LIBERO](https://github.com/Lifelong-Robot-Learning/LIBERO) benchmark, fine-tuned by Advanced Micro Devices, Inc. from [Wan2.2-TI2V-5B](https://huggingface.co/Wan-AI/Wan2.2-TI2V-5B).
17
 
18
  Code: [AMD-AGI/faster-action-models](https://github.com/AMD-AGI/faster-action-models).
19
- Write-up and reference results: [Optimizing World Action Models on AMD GPUs](https://rocm.blogs.amd.com/artificial-intelligence/faster-action-models/README.html).
20
 
21
  Part of the [Compact Action Expert (CAE) collection](https://huggingface.co/collections/amd/compact-action-expert-cae-6ace112e7bd665ff8c107ed2).
22
 
@@ -26,16 +26,16 @@ The Compact Action Expert combines a 32-dimensional action expert (about 30M par
26
 
27
  ## Results
28
 
29
- LIBERO (all four suites, 50 episodes per task) on one AMD Instinct™ MI355X GPU with ROCm 7.14.0 and PyTorch 2.11.0, simulated on the GPU (`--sim instinct`), with the GEAK-optimized inference path from the code repository. Reference results, from Table 3 of the [write-up](https://rocm.blogs.amd.com/artificial-intelligence/faster-action-models/README.html):
30
 
31
- | Configuration | Success (%) | Latency per policy decision (ms) |
32
  |---|---|---|
33
  | FastWAM, no optimization (10 denoising steps) | 96.95 ± 0.22 | 222.55 |
34
  | FastWAM + GEAK (10 steps) | 96.86 ± 0.30 | 24.17 |
35
  | CAE-FastWAM + GEAK (10 steps) | 97.28 ± 0.20 | 12.82 |
36
  | CAE-FastWAM + GEAK (4 steps) | 97.33 ± 0.26 | 8.83 |
37
 
38
- Success is the mean ± standard deviation over 11 full LIBERO runs (5 for the 4-step row). The FastWAM rows use the authors' [`yuanty/fastwam`](https://huggingface.co/yuanty/fastwam) checkpoint. Reproduce the table with `scripts/reproduce_cae_fastwam.sh` in the code repository. Training took 22.7 hours on two AMD Instinct™ MI325X GPUs. Results are simulator measurements and may vary with configuration and software versions.
39
 
40
  ## Files
41
 
 
16
  A latent world-action policy for the [LIBERO](https://github.com/Lifelong-Robot-Learning/LIBERO) benchmark, fine-tuned by Advanced Micro Devices, Inc. from [Wan2.2-TI2V-5B](https://huggingface.co/Wan-AI/Wan2.2-TI2V-5B).
17
 
18
  Code: [AMD-AGI/faster-action-models](https://github.com/AMD-AGI/faster-action-models).
19
+ ROCm blog: [Optimizing World Action Models on AMD GPUs](https://rocm.blogs.amd.com/artificial-intelligence/faster-action-models/README.html).
20
 
21
  Part of the [Compact Action Expert (CAE) collection](https://huggingface.co/collections/amd/compact-action-expert-cae-6ace112e7bd665ff8c107ed2).
22
 
 
26
 
27
  ## Results
28
 
29
+ LIBERO (all four suites, 50 episodes per task) on one AMD Instinct™ MI355X GPU with ROCm 7.14.0 and PyTorch 2.11.0, simulated on the GPU (`--sim instinct`), with the GEAK-optimized inference path from the code repository. Reference results, from Table 3 of the [ROCm blog](https://rocm.blogs.amd.com/artificial-intelligence/faster-action-models/README.html):
30
 
31
+ | Configuration | Success rate (%) | Latency per policy decision (ms) |
32
  |---|---|---|
33
  | FastWAM, no optimization (10 denoising steps) | 96.95 ± 0.22 | 222.55 |
34
  | FastWAM + GEAK (10 steps) | 96.86 ± 0.30 | 24.17 |
35
  | CAE-FastWAM + GEAK (10 steps) | 97.28 ± 0.20 | 12.82 |
36
  | CAE-FastWAM + GEAK (4 steps) | 97.33 ± 0.26 | 8.83 |
37
 
38
+ Success rate is the mean ± standard deviation over 11 full LIBERO runs (5 for the 4-step row). The FastWAM rows use the authors' [`yuanty/fastwam`](https://huggingface.co/yuanty/fastwam) checkpoint. Reproduce the table with `scripts/reproduce_cae_fastwam.sh` in the code repository. Training took 22.7 hours on two AMD Instinct™ MI325X GPUs. Results are simulator measurements and may vary with configuration and software versions.
39
 
40
  ## Files
41