diff --git a/.riahub/train_configs/train.yaml b/.riahub/train_configs/train.yaml index dbfc47d..6115236 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: false + drop_last: true kind: iq_h5_auto num_workers: 4 persistent_workers: false pin_memory: true - test_path: /opt/qmb/riahub/dataset/qoherent/RIA_Example/main/Datasets/test.h5 + test_path: /opt/qmb/riahub/dataset/qoherent/pluto-interference-recognition-demo/main/test.h5 test_split: 0 - train_path: /opt/qmb/riahub/dataset/qoherent/RIA_Example/main/Datasets/train.h5 + train_path: /opt/qmb/riahub/dataset/qoherent/pluto-interference-recognition-demo/main/train.h5 validation_split: 0 evaluation: capture_predictions: true @@ -25,7 +25,7 @@ export: dynamic_width: false enabled: true file_name: model.onnx - opset_version: 17 + opset_version: 23 strict: false use_dynamo: true use_onnxsim: false diff --git a/.riahub/workflows/train.yaml b/.riahub/workflows/train.yaml index 5b8d577..43dcd7d 100644 --- a/.riahub/workflows/train.yaml +++ b/.riahub/workflows/train.yaml @@ -39,7 +39,7 @@ jobs: fi - - name: Checkout Datasets (qoherent/RIA_Example) + - name: Checkout Datasets (qoherent/pluto-interference-recognition-demo) env: RIAHUB_USER: ${{ secrets.QMBDEMO_USER }} RIAHUB_TOKEN: ${{ secrets.QMBDEMO_TOKEN }} @@ -64,8 +64,8 @@ jobs: echo "http://$raw" } - REPO_PATH="/qoherent/RIA_Example.git" - DEST_ROOT="/opt/qmb/riahub/dataset/qoherent/RIA_Example/main" + REPO_PATH="/qoherent/pluto-interference-recognition-demo.git" + DEST_ROOT="/opt/qmb/riahub/dataset/qoherent/pluto-interference-recognition-demo/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 -- \ - "Datasets/test.h5" \ - "Datasets/train.h5" - sudo git -C "$DEST_ROOT" fetch --depth=1 origin "120041747ecb706fe9a3526f3e10f3e4932b6aee" + "test.h5" \ + "train.h5" + sudo git -C "$DEST_ROOT" fetch --depth=1 origin "fca8c5d477b5b20a9b2678404b070d7ac90587a6" sudo git -C "$DEST_ROOT" -c advice.detachedHead=false checkout FETCH_HEAD - sudo git -C "$DEST_ROOT" lfs fetch origin --include="Datasets/test.h5,Datasets/train.h5" --exclude="" || true + sudo git -C "$DEST_ROOT" lfs fetch origin --include="test.h5,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:52:55.538324+00:00 +# committed at 2026-05-26T16:56:14.829285+00:00