Unverified 提交 34b859a4 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Keras CI fix (#5486)

* Keras CI fix * pre-commit fixes * Update ci-testing.yml Co-authored-by: 's avatarpre-commit <pre-commit@example.com>
上级 06bf8ef7
...@@ -46,11 +46,12 @@ jobs: ...@@ -46,11 +46,12 @@ jobs:
restore-keys: | restore-keys: |
${{ runner.os }}-${{ matrix.python-version }}-pip- ${{ runner.os }}-${{ matrix.python-version }}-pip-
# Known Keras 2.7.0 issue: https://github.com/ultralytics/yolov5/pull/5486
- name: Install dependencies - name: Install dependencies
run: | run: |
python -m pip install --upgrade pip python -m pip install --upgrade pip
pip install -qr requirements.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html 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 python --version
pip --version pip --version
pip list pip list
......
...@@ -1085,4 +1085,4 @@ ...@@ -1085,4 +1085,4 @@
"outputs": [] "outputs": []
} }
] ]
} }
\ No newline at end of file
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论