Model card: link the ROCm blog, label success rate
#2
by Viani - opened
- 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 |
|