Browse Source

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 3 years ago
parent
commit
34b859a41e
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
  1. +2
    -1
      .github/workflows/ci-testing.yml
  2. +1
    -1
      tutorial.ipynb

+ 2
- 1
.github/workflows/ci-testing.yml View File

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

+ 1
- 1
tutorial.ipynb View File

"outputs": [] "outputs": []
} }
] ]
}
}

Loading…
Cancel
Save