diff --git a/.riahub/train_configs/model/model.yaml b/.riahub/train_configs/model/model.yaml
index 4183b59..f192199 100644
--- a/.riahub/train_configs/model/model.yaml
+++ b/.riahub/train_configs/model/model.yaml
@@ -1,5 +1,5 @@
 source: wavesfm
 name: WavesFM Linear Probe
 task: rml
-epochs: 10
+epochs: 5
 batch_size: 2048
diff --git a/.riahub/workflows/train.yaml b/.riahub/workflows/train.yaml
index 938f513..ff464be 100644
--- a/.riahub/workflows/train.yaml
+++ b/.riahub/workflows/train.yaml
@@ -15,10 +15,10 @@ permissions:
 
 jobs:
   WavesFM-Training:
-    runs-on: "ubuntu-24.04"
+    runs-on: "ubuntu-latest-2080"
     env:
       WAVESFM_TASK: "rml"
-      WAVESFM_EPOCHS: "10"
+      WAVESFM_EPOCHS: "5"
       WAVESFM_BATCH_SIZE: "2048"
       WAVESFM_OUTPUT_DIR: "/opt/wavesfm/output"
       # Single source of truth for the cloned WavesFM repo location.
@@ -472,4 +472,4 @@ jobs:
             ${{ env.WAVESFM_OUTPUT_DIR }}/best.pth
             ${{ env.WAVESFM_OUTPUT_DIR }}/log.txt
           if-no-files-found: warn
-# committed at 2026-05-28T11:57:55.094537+00:00
+# committed at 2026-05-28T15:50:37.929667+00:00