check_requirements(('coremltools',)) (#4478)
* `check_requirements(('coremltools',))`
* Update ci-testing.yml
* Update ci-testing.yml
This commit is contained in:
1 parent
1d65e8194d
commit
d1182c4f29
2 files changed
+3
-2
No files matched your search
@@ -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
|
||||
Reference in new issue
Block a user