diff --git a/.riahub/train_configs/train.yaml b/.riahub/train_configs/train.yaml
index e98affb..dbfc47d 100644
--- a/.riahub/train_configs/train.yaml
+++ b/.riahub/train_configs/train.yaml
@@ -5,14 +5,14 @@ data:
         label_key: impairment
         metadata_key: metadata/metadata
         snr_key: Z
-    drop_last: true
+    drop_last: false
     kind: iq_h5_auto
     num_workers: 4
     persistent_workers: false
     pin_memory: true
-    test_path: /opt/qmb/riahub/dataset/qoherent/pluto-interference-recognition-demo/main/test.h5
+    test_path: /opt/qmb/riahub/dataset/qoherent/RIA_Example/main/Datasets/test.h5
     test_split: 0
-    train_path: /opt/qmb/riahub/dataset/qoherent/pluto-interference-recognition-demo/main/train.h5
+    train_path: /opt/qmb/riahub/dataset/qoherent/RIA_Example/main/Datasets/train.h5
     validation_split: 0
 evaluation:
     capture_predictions: true
@@ -45,12 +45,12 @@ optimization:
             lr: 0.001
             weight_decay: 0
 runtime:
-    amp_enabled: false
-    autocast_dtype: float32
+    amp_enabled: true
+    autocast_dtype: float16
     checkpoint_every_n_epochs: 1
     component_modules: []
     device: auto
-    epochs: 1
+    epochs: 10
     progress_bar: false
     seed: 42
 task:
diff --git a/.riahub/workflows/train.yaml b/.riahub/workflows/train.yaml
index b665451..5b8d577 100644
--- a/.riahub/workflows/train.yaml
+++ b/.riahub/workflows/train.yaml
@@ -39,7 +39,7 @@ jobs:
           fi
 
 
-      - name: Checkout Datasets (qoherent/pluto-interference-recognition-demo)
+      - name: Checkout Datasets (qoherent/RIA_Example)
         env:
           RIAHUB_USER: ${{ secrets.QMBDEMO_USER }}
           RIAHUB_TOKEN: ${{ secrets.QMBDEMO_TOKEN }}
@@ -64,8 +64,8 @@ jobs:
             echo "http://$raw"
           }
 
-          REPO_PATH="/qoherent/pluto-interference-recognition-demo.git"
-          DEST_ROOT="/opt/qmb/riahub/dataset/qoherent/pluto-interference-recognition-demo/main"
+          REPO_PATH="/qoherent/RIA_Example.git"
+          DEST_ROOT="/opt/qmb/riahub/dataset/qoherent/RIA_Example/main"
           sudo mkdir -p "$(dirname "$DEST_ROOT")"
           mapfile -t BASE_CANDIDATES < <(build_base_candidates "$BASE_URL_SOURCE")
           CLONED=0
@@ -91,11 +91,11 @@ jobs:
           sudo git -C "$DEST_ROOT" lfs install --local || true
           sudo git -C "$DEST_ROOT" sparse-checkout init --no-cone
           sudo git -C "$DEST_ROOT" sparse-checkout set --no-cone -- \
-            "test.h5" \
-            "train.h5"
-          sudo git -C "$DEST_ROOT" fetch --depth=1 origin "fca8c5d477b5b20a9b2678404b070d7ac90587a6"
+            "Datasets/test.h5" \
+            "Datasets/train.h5"
+          sudo git -C "$DEST_ROOT" fetch --depth=1 origin "120041747ecb706fe9a3526f3e10f3e4932b6aee"
           sudo git -C "$DEST_ROOT" -c advice.detachedHead=false checkout FETCH_HEAD
-          sudo git -C "$DEST_ROOT" lfs fetch origin --include="test.h5,train.h5" --exclude="" || true
+          sudo git -C "$DEST_ROOT" lfs fetch origin --include="Datasets/test.h5,Datasets/train.h5" --exclude="" || true
           sudo git -C "$DEST_ROOT" lfs checkout || true
           sudo git -C "$DEST_ROOT" remote remove origin || true
           sudo git -C "$DEST_ROOT" config --local --unset-all http.extraheader || true
@@ -198,4 +198,4 @@ jobs:
           path: ${{ github.workspace }}/.riahub/artifacts/training
           if-no-files-found: warn
 
-# committed at 2026-05-26T16:02:23.409072+00:00
+# committed at 2026-05-26T16:52:55.538324+00:00