From 34b859a41e7a97ee519e08523f67a50f5a12dacb Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 3 Nov 2021 19:25:44 +0100 Subject: [PATCH] Keras CI fix (#5486) * Keras CI fix * pre-commit fixes * Update ci-testing.yml Co-authored-by: pre-commit --- .github/workflows/ci-testing.yml | 3 ++- tutorial.ipynb | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml index 8ebfdec..abfe21e 100644 --- a/.github/workflows/ci-testing.yml +++ b/.github/workflows/ci-testing.yml @@ -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 diff --git a/tutorial.ipynb b/tutorial.ipynb index b81db0d..9184a66 100644 --- a/tutorial.ipynb +++ b/tutorial.ipynb @@ -1085,4 +1085,4 @@ "outputs": [] } ] -} \ No newline at end of file +}