`check_requirements(('coremltools',))` (#4478)
* `check_requirements(('coremltools',))` * Update ci-testing.yml * Update ci-testing.ymlpull/4481/head
parent
1d65e8194d
commit
d1182c4f29
|
@ -48,7 +48,7 @@ jobs:
|
|||
run: |
|
||||
python -m pip install --upgrade pip
|
||||
pip install -qr requirements.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html
|
||||
pip install -q onnx
|
||||
pip install -q onnx onnx-simplifier coremltools # for export
|
||||
python --version
|
||||
pip --version
|
||||
pip list
|
||||
|
@ -76,5 +76,5 @@ jobs:
|
|||
|
||||
python hubconf.py # hub
|
||||
python models/yolo.py --cfg ${{ matrix.model }}.yaml # inspect
|
||||
python export.py --img 128 --batch 1 --weights ${{ matrix.model }}.pt # export
|
||||
python export.py --img 128 --batch 1 --weights ${{ matrix.model }}.pt --include onnx torchscript # export
|
||||
shell: bash
|
||||
|
|
Loading…
Reference in New Issue