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

disabled gpu for now
Some checks failed
RIA Hub Workflow Demo / ria-demo (push) Failing after 35s
Details

Browse Source
This commit is contained in:
liyuxiao2 2025-05-22 14:33:08 -04:00
parent 310cc10f71
commit 85d6afd976
4 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.riahub/workflows/workflow.yaml
View File

@ -46,6 +46,7 @@ jobs:
- name: 2. Train Model - name: 2. Train Model
run: | run: |
mkdir -p data/dataset
PYTHONPATH=. python data/training/train.py PYTHONPATH=. python data/training/train.py
echo "training model" echo "training model"

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: results/inference_recognition_model.onnx output_path: onnx_files/inference_recognition_model.onnx
app: app:
build_dir: dist build_dir: dist

2
data/training/train.py
View File

@ -141,7 +141,7 @@ def train_model():
trainer = L.Trainer( trainer = L.Trainer(
max_epochs=epochs, max_epochs=epochs,
callbacks=[checkpoint_callback], callbacks=[checkpoint_callback],
accelerator="gpu", accelerator="cpu",
devices=1, devices=1,
benchmark=True, benchmark=True,
precision="bf16-mixed", precision="bf16-mixed",

BIN
results/interference_recognition_model.ckpt
View File

Binary file not shown.