Keras CI fix (#5486)
* Keras CI fix * pre-commit fixes * Update ci-testing.yml Co-authored-by: pre-commit <pre-commit@example.com>pull/5487/head
parent
06bf8ef7e5
commit
34b859a41e
|
@ -46,11 +46,12 @@ jobs:
|
|||
restore-keys: |
|
||||
${{ runner.os }}-${{ matrix.python-version }}-pip-
|
||||
|
||||
# Known Keras 2.7.0 issue: https://github.com/ultralytics/yolov5/pull/5486
|
||||
- name: Install dependencies
|
||||
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 tensorflow-cpu # for export
|
||||
pip install -q onnx tensorflow-cpu keras==2.6.0 # for export
|
||||
python --version
|
||||
pip --version
|
||||
pip list
|
||||
|
|
|
@ -1085,4 +1085,4 @@
|
|||
"outputs": []
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue