diff --git a/.riahub/workflows/workflow.yaml b/.riahub/workflows/workflow.yaml index de58f39..0d04c19 100644 --- a/.riahub/workflows/workflow.yaml +++ b/.riahub/workflows/workflow.yaml @@ -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 diff --git a/scripts/ort/convert_to_ort.py b/scripts/ort/convert_to_ort.py index 9593f46..8d44f5a 100644 --- a/scripts/ort/convert_to_ort.py +++ b/scripts/ort/convert_to_ort.py @@ -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