qoherent/modrec-workflow
14
0
Fork 1
Code Issues 1 Pull Requests Workflows Packages Projects Releases 1 Wiki Activity

fixed paths
Some checks failed
RIA Hub Workflow Demo / ria-demo (push) Failing after 28s
Details

Browse Source
This commit is contained in:
liyuxiao2 2025-05-22 14:26:15 -04:00
parent dcb6e0accd
commit 5970af5be7
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
conf/app.yaml
View File

@ -21,7 +21,7 @@ training:
inference: inference:
model_path: checkpoints/inference_recognition_model.ckpt model_path: checkpoints/inference_recognition_model.ckpt
num_classes: 4 num_classes: 4
output_path: onnx_files/inference_recognition_model.onnx output_path: results/inference_recognition_model.onnx
app: app:
build_dir: dist build_dir: dist

9
data/training/train.py
View File

@ -26,19 +26,20 @@ import mobilenetv3
def train_model(): def train_model():
settings = get_app_settings() settings = get_app_settings()
dataset = settings.dataset.modulation_types training_cfg = settings.training
dataset_cfg = settings.dataset
train_flag = True train_flag = True
batch_size = 128 batch_size = 128
epochs = 1 epochs = 1
checkpoint_filename = f"/Users/liyuxiao/Documents/CS/qoherent/modrec-workflow/results/interference_recognition_model" checkpoint_filename = f"{training_cfg.checkpoint_path}"
train_data = ( train_data = (
"/Users/liyuxiao/Documents/CS/qoherent/modrec-workflow/data/dataset/train.h5" f"{settings.dataset_cfg.output_dir}/train.h5"
) )
val_data = ( val_data = (
"/Users/liyuxiao/Documents/CS/qoherent/modrec-workflow/data/dataset/val.h5" f"{settings.dataset_cfg.output_dir}/val.h5"
) )
dataset_name = "Modulation Inference - Initial Model" dataset_name = "Modulation Inference - Initial Model"