Robotics
PyTorch
world-action-model
libero
Viani commited on
Commit
a600ee3
·
verified ·
1 Parent(s): 363bdc0

Model card: link the ROCm blog, label success rate

Browse files

Refer to the ROCm blog by name and label the results column and note as success rate. Updates the README.md line in ASSET_SHA256SUMS. No other file changes.

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