diff --git a/.riahub/workflows/workflow.yaml b/.riahub/workflows/workflow.yaml index aab0ee2..d52c5e3 100644 --- a/.riahub/workflows/workflow.yaml +++ b/.riahub/workflows/workflow.yaml @@ -78,9 +78,7 @@ jobs: name: checkpoints path: checkpoint_files/* - - name: List checkpoint directory - run: ls -lh checkpoint_files - + - name: 4. Convert to ONNX file run: | mkdir -p onnx_files @@ -93,6 +91,10 @@ jobs: name: ria-demo-onnx path: onnx_files/inference_recognition_model.onnx + - name: List checkpoint directory + run: ls -lh checkpoint_files + + - name: 5. Profile ONNX model run: | PYTHONPATH=. python onnx_scripts/profile_onnx.py