michael/modrec-workflow
1
0
Fork 0
forked from qoherent/modrec-workflow
Code Pull Requests Workflows Activity

updated workflow scripts

Browse Source
This commit is contained in:
liyuxiao2 2025-05-21 15:46:59 -04:00
parent ba7d0d9f67
commit b49f351a4c
3 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.riahub/workflows/workflow.yaml
View File

@ -40,7 +40,7 @@ jobs:
- name: 1. Build HDF5 Dataset - name: 1. Build HDF5 Dataset
run: | run: |
mkdir -p data mkdir -p data
PYTHONPATH=. python scripts/produce_dataset.py PYTHONPATH=. python data/scripts/produce_dataset.py
echo "datasets produced successfully" echo "datasets produced successfully"
shell: bash shell: bash

15
conf/app.yaml
View File

@ -3,24 +3,25 @@ general:
run_mode: prod run_mode: prod
dataset: dataset:
input_dir: "recordings" input_dir: data/recordings
num_slices: 8 num_slices: 8
train_split: 0.8 train_split: 0.8
val_split : 0.2 val_split : 0.2
seed: 42 seed: 25
output_dir: "data" modulation_types: [bpsk, qpsk, qam16, qam64]
output_dir: data/dataset
training: training:
batch_size: 64 batch_size: 64
epochs: 50 epochs: 50
learning_rate: 0.001 learning_rate: 0.001
checkpoint_path: "checkpoints/model.ckpt" checkpoint_path: checkpoints/model.ckpt
use_gpu: true use_gpu: true
inference: inference:
model_path: "checkpoints/model.ckpt" model_path: checkpoints/model.ckpt
num_classes: 10 num_classes: 10
output_path: "results/output.json" output_path: results/output.json
app: app:
build_dir: "dist" build_dir: dist

1
helpers/app_settings.py
View File

@ -14,6 +14,7 @@ class DataSetConfig:
num_slices: int num_slices: int
train_split: float train_split: float
seed: int seed: int
modulation_types: list
val_split: float val_split: float
output_dir: str output_dir: str