Skip to content

Commit 1f2f6de

Browse files
glenn-jocherpre-commit
andauthored
Keras CI fix (ultralytics#5486)
* Keras CI fix * pre-commit fixes * Update ci-testing.yml Co-authored-by: pre-commit <[email protected]>
1 parent b751f34 commit 1f2f6de

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

.github/workflows/ci-testing.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,12 @@ jobs:
4646
restore-keys: |
4747
${{ runner.os }}-${{ matrix.python-version }}-pip-
4848
49+
# Known Keras 2.7.0 issue: https://github.com/ultralytics/yolov5/pull/5486
4950
- name: Install dependencies
5051
run: |
5152
python -m pip install --upgrade pip
5253
pip install -qr requirements.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html
53-
pip install -q onnx tensorflow-cpu # for export
54+
pip install -q onnx tensorflow-cpu keras==2.6.0 # for export
5455
python --version
5556
pip --version
5657
pip list

tutorial.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1085,4 +1085,4 @@
10851085
"outputs": []
10861086
}
10871087
]
1088-
}
1088+
}

0 commit comments

Comments
 (0)