|
|
|
@ -107,9 +107,9 @@ jobs:
|
|
|
|
|
run: |
|
|
|
|
|
python -m pip install --upgrade pip wheel
|
|
|
|
|
if [ "${{ matrix.os }}" == "macos-latest" ]; then
|
|
|
|
|
pip install -e '.[export]' --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
|
pip install -e ".[export]" --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
|
else
|
|
|
|
|
pip install -e '.[export]' --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
|
pip install -e ".[export]" --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
|
fi
|
|
|
|
|
yolo export format=tflite imgsz=32
|
|
|
|
|
- name: Check environment
|
|
|
|
@ -165,9 +165,9 @@ jobs:
|
|
|
|
|
run: | # CoreML must be installed before export due to protobuf error from AutoInstall
|
|
|
|
|
python -m pip install --upgrade pip wheel
|
|
|
|
|
if [ "${{ matrix.torch }}" == "1.8.0" ]; then
|
|
|
|
|
pip install -e . torch==1.8.0 torchvision==0.9.0 pytest 'coremltools>=6.0,<=6.2' --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
|
pip install -e . torch==1.8.0 torchvision==0.9.0 pytest "coremltools>=6.0,<=6.2" --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
|
else
|
|
|
|
|
pip install -e . pytest 'coremltools>=6.0,<=6.2' --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
|
pip install -e . pytest "coremltools>=6.0,<=6.2" --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
|
fi
|
|
|
|
|
- name: Check environment
|
|
|
|
|
run: |
|
|
|
|
|