apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update ci-testing.yml

Browse Source
This commit is contained in:
Glenn Jocher 2022-10-16 11:45:19 +02:00 committed by GitHub
parent b0b071dda8
commit 88a803126b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.github/workflows/ci-testing.yml vendored
View File

@ -51,7 +51,6 @@ jobs:
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 keras==2.6.0 # wandb # extras
python --version
pip --version
pip list
@ -83,7 +82,7 @@ jobs:
# Export
python models/yolo.py --cfg ${{ matrix.model }}.yaml # build PyTorch model
# python models/tf.py --weights ${{ matrix.model }}.pt # build TensorFlow model (YOLOv3 not supported)
python export.py --img 64 --batch 1 --weights runs/train/exp/weights/last.pt --include torchscript onnx # export
python export.py --img 64 --batch 1 --weights runs/train/exp/weights/last.pt --include torchscript # export
# Python
python - <<EOF
import torch