Model card: link the ROCm blog, label success rate
Browse filesRefer 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.
- ASSET_SHA256SUMS +1 -1
- 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 |
-
|
| 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 |
-
|
| 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 [
|
| 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 |
|