Documentation and formatting updates #1
|
@ -1,4 +1,4 @@
|
||||||
name: RIA Hub Workflow Demo
|
name: Modulation Recognition Demo
|
||||||
|
|
||||||
on:
|
on:
|
||||||
push:
|
push:
|
||||||
|
@ -11,9 +11,6 @@ on:
|
||||||
jobs:
|
jobs:
|
||||||
ria-demo:
|
ria-demo:
|
||||||
runs-on: ubuntu-latest-2080
|
runs-on: ubuntu-latest-2080
|
||||||
env:
|
|
||||||
RIAGIT_USERNAME: ${{ secrets.USERNAME }}
|
|
||||||
RIAGIT_TOKEN: ${{ secrets.TOKEN }}
|
|
||||||
steps:
|
steps:
|
||||||
- name: Print GPU information
|
- name: Print GPU information
|
||||||
run: |
|
run: |
|
||||||
|
@ -24,7 +21,7 @@ jobs:
|
||||||
echo "⚠️ No NVIDIA GPU found"
|
echo "⚠️ No NVIDIA GPU found"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
- name: Checkout code
|
- name: Checkout project code
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
lfs: true
|
lfs: true
|
||||||
|
@ -42,13 +39,10 @@ jobs:
|
||||||
utils \
|
utils \
|
||||||
-r requirements.txt
|
-r requirements.txt
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
- name: 1. Generate Recordings
|
- name: 1. Generate Recordings
|
||||||
run: |
|
run: |
|
||||||
mkdir -p data/recordings
|
mkdir -p data/recordings
|
||||||
PYTHONPATH=. python scripts/dataset_building/data_gen.py --output-dir data/recordings
|
PYTHONPATH=. python scripts/dataset_manager/data_gen.py --output-dir data/recordings
|
||||||
echo "recordings produced successfully"
|
|
||||||
|
|
||||||
- name: ⬆️ Upload recordings
|
- name: ⬆️ Upload recordings
|
||||||
uses: actions/upload-artifact@v3
|
uses: actions/upload-artifact@v3
|
||||||
|
@ -59,11 +53,10 @@ jobs:
|
||||||
- name: 2. Build HDF5 Dataset
|
- name: 2. Build HDF5 Dataset
|
||||||
run: |
|
run: |
|
||||||
mkdir -p data/dataset
|
mkdir -p data/dataset
|
||||||
PYTHONPATH=. python scripts/dataset_building/produce_dataset.py
|
PYTHONPATH=. python scripts/dataset_manager/produce_dataset.py
|
||||||
echo "datasets produced successfully"
|
|
||||||
shell: bash
|
shell: bash
|
||||||
|
|
||||||
- name: 📤 Upload Dataset
|
- name: ⬆️ Upload Dataset
|
||||||
uses: actions/upload-artifact@v3
|
uses: actions/upload-artifact@v3
|
||||||
with:
|
with:
|
||||||
name: dataset
|
name: dataset
|
||||||
|
@ -75,34 +68,30 @@ jobs:
|
||||||
PYTORCH_NO_NNPACK: 1
|
PYTORCH_NO_NNPACK: 1
|
||||||
run: |
|
run: |
|
||||||
mkdir -p checkpoint_files
|
mkdir -p checkpoint_files
|
||||||
PYTHONPATH=. python scripts/training/train.py 2>/dev/null
|
PYTHONPATH=. python scripts/model_builder/train.py 2>/dev/null
|
||||||
echo "training model"
|
|
||||||
|
|
||||||
- name: 4. Plot Model
|
- name: 4. Plot Model
|
||||||
env:
|
env:
|
||||||
NO_NNPACK: 1
|
NO_NNPACK: 1
|
||||||
PYTORCH_NO_NNPACK: 1
|
PYTORCH_NO_NNPACK: 1
|
||||||
run: |
|
run: |
|
||||||
PYTHONPATH=. python scripts/training/plot_data.py 2>/dev/null
|
PYTHONPATH=. python scripts/model_builder/plot_data.py 2>/dev/null
|
||||||
|
|
||||||
|
|
||||||
- name: Upload Checkpoints
|
- name: ⬆️ Upload Checkpoints
|
||||||
uses: actions/upload-artifact@v3
|
uses: actions/upload-artifact@v3
|
||||||
with:
|
with:
|
||||||
name: checkpoints
|
name: checkpoints
|
||||||
path: checkpoint_files/*
|
path: checkpoint_files/*
|
||||||
|
|
||||||
|
- name: 5. Export model to ONNX graph
|
||||||
- name: 5. Convert to ONNX file
|
|
||||||
env:
|
env:
|
||||||
NO_NNPACK: 1
|
NO_NNPACK: 1
|
||||||
PYTORCH_NO_NNPACK: 1
|
PYTORCH_NO_NNPACK: 1
|
||||||
run: |
|
run: |
|
||||||
mkdir -p onnx_files
|
mkdir -p onnx_files
|
||||||
MKL_DISABLE_FAST_MM=1 PYTHONPATH=. python scripts/onnx/convert_to_onnx.py 2>/dev/null
|
MKL_DISABLE_FAST_MM=1 PYTHONPATH=. python scripts/application_packager/convert_to_onnx.py 2>/dev/null
|
||||||
echo "building inference app"
|
|
||||||
|
|
||||||
- name: Upload ONNX file
|
- name: ⬆️ Upload ONNX file
|
||||||
uses: actions/upload-artifact@v3
|
uses: actions/upload-artifact@v3
|
||||||
with:
|
with:
|
||||||
name: onnx-file
|
name: onnx-file
|
||||||
|
@ -110,21 +99,20 @@ jobs:
|
||||||
|
|
||||||
- name: 6. Profile ONNX model
|
- name: 6. Profile ONNX model
|
||||||
run: |
|
run: |
|
||||||
PYTHONPATH=. python scripts/onnx/profile_onnx.py
|
PYTHONPATH=. python scripts/application_packager/profile_onnx.py
|
||||||
|
|
||||||
- name: Upload JSON profiling data
|
- name: ⬆️ Upload JSON trace
|
||||||
uses: actions/upload-artifact@v3
|
uses: actions/upload-artifact@v3
|
||||||
with:
|
with:
|
||||||
name: profile-data
|
name: profile-data
|
||||||
path: '**/onnxruntime_profile_*.json'
|
path: '**/onnxruntime_profile_*.json'
|
||||||
|
|
||||||
- name: 7. Convert to ORT file
|
- name: 7. Convert ONNX graph to an ORT file
|
||||||
run: |
|
run: |
|
||||||
PYTHONPATH=. python scripts/ort/convert_to_ort.py
|
PYTHONPATH=. python scripts/application_packager/convert_to_ort.py
|
||||||
|
|
||||||
- name: Upload ORT file
|
- name: ⬆️ Upload ORT file
|
||||||
uses: actions/upload-artifact@v3
|
uses: actions/upload-artifact@v3
|
||||||
with:
|
with:
|
||||||
name: ort-file
|
name: ort-file
|
||||||
path: ort_files/inference_recognition_model.ort
|
path: ort_files/inference_recognition_model.ort
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user