Selaa lähdekoodia

Keras CI fix (#5486)

* Keras CI fix

* pre-commit fixes

* Update ci-testing.yml

Co-authored-by: pre-commit <pre-commit@example.com>
modifyDataloader
Glenn Jocher GitHub 2 vuotta sitten
vanhempi
commit
34b859a41e
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
2 muutettua tiedostoa jossa 3 lisäystä ja 2 poistoa
  1. +2
    -1
      .github/workflows/ci-testing.yml
  2. +1
    -1
      tutorial.ipynb

+ 2
- 1
.github/workflows/ci-testing.yml Näytä tiedosto

@@ -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

+ 1
- 1
tutorial.ipynb Näytä tiedosto

@@ -1085,4 +1085,4 @@
"outputs": []
}
]
}
}

Loading…
Peruuta
Tallenna