Skip to content

add onnx check

add onnx check #14

Workflow file for this run

name: CI
on:
push:
branches: ["main"]
pull_request:
branches: ["main"]
jobs:
build:
runs-on: self-hosted
steps:
- uses: actions/checkout@v3
- uses: actions/setup-python@v4
with:
python-version: "3.10"
- name: model zoo test
run: |
python -m pip install --upgrade pip wheel setuptools
pip install .
pip install pytest onnxruntime
python tests/test_folder.py --model-dir /root/data/modelzoo
- name: pattern matcher test
run: |
pytest tests/test_pattern_matcher.py
- name: pattern generator test
run: |
pytest tests/test_pattern_generator.py
- name: yolo test
run: |
pip install pytest pytest-xdist onnxruntime timm torchvision --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu
pip install ultralytics
pytest tests/test_yolo.py