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

added warning dumps

Browse Source
This commit is contained in:
Liyu Xiao 2025-06-18 13:59:48 -04:00
parent 42e7429f02
commit 9f4b71437e
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
.riahub/workflows/workflow.yaml
View File

@ -71,7 +71,7 @@ jobs:
PYTORCH_NO_NNPACK: 1
run: |
mkdir -p checkpoint_files
PYTHONPATH=. python scripts/training/train.py
PYTHONPATH=. python scripts/training/train.py 2>/dev/null
echo "training model"
- name: 4. Plot Model
@ -79,7 +79,7 @@ jobs:
NO_NNPACK: 1
PYTORCH_NO_NNPACK: 1
run: |
PYTHONPATH=. python scripts/training/plot_data.py
PYTHONPATH=. python scripts/training/plot_data.py 2>/dev/null
- name: Upload Checkpoints
@ -95,7 +95,7 @@ jobs:
PYTORCH_NO_NNPACK: 1
run: |
mkdir -p onnx_files
MKL_DISABLE_FAST_MM=1 PYTHONPATH=. python scripts/onnx/convert_to_onnx.py
MKL_DISABLE_FAST_MM=1 PYTHONPATH=. python scripts/onnx/convert_to_onnx.py 2>/dev/null
echo "building inference app"
- name: Upload ONNX file

2
scripts/ort/convert_to_ort.py
View File

@ -3,7 +3,7 @@ from helpers.app_settings import get_app_settings
settings = get_app_settings()
input_path = f"{settings.app.build_dir}/inference_recognition_model.onnx"
input_path = "onnx_files/inference_recognition_model.onnx"
optimization_style = settings.app.optimization_style