diff --git a/.riahub/train_configs/model/model.yaml b/.riahub/train_configs/model/model.yaml
index 4183b59..6a0fae1 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: 4
 batch_size: 2048
diff --git a/.riahub/workflows/train.yaml b/.riahub/workflows/train.yaml
index 40eead6..d2b092b 100644
--- a/.riahub/workflows/train.yaml
+++ b/.riahub/workflows/train.yaml
@@ -15,10 +15,10 @@ permissions:
 
 jobs:
   WavesFM-Training:
-    runs-on: "ubuntu-latest-2080"
+    runs-on: "ubuntu-24.04"
     env:
       WAVESFM_TASK: "rml"
-      WAVESFM_EPOCHS: "10"
+      WAVESFM_EPOCHS: "4"
       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:10:29.803581+00:00
+# committed at 2026-05-28T11:12:10.069174+00:00