Просмотр исходного кода

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 лет назад
Родитель
Сommit
34b859a41e
Не найден GPG ключ соответствующий данной подписи Идентификатор GPG ключа: 4AEE18F83AFDEB23
2 измененных файлов: 3 добавлений и 2 удалений
  1. +2
    -1
      .github/workflows/ci-testing.yml
  2. +1
    -1
      tutorial.ipynb

+ 2
- 1
.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

+ 1
- 1
tutorial.ipynb Просмотреть файл

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

Загрузка…
Отмена
Сохранить