Skip to content

Commit

Permalink
Add PyTorch Hub classification CI checks
Browse files Browse the repository at this point in the history
Add PyTorch Hub loading of official and custom trained classification models to CI checks. 

May help resolve #8790 (comment)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
  • Loading branch information
glenn-jocher committed Aug 18, 2022
1 parent a5a47c5 commit 8fd337b
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions .github/workflows/ci-testing.yml
Original file line number Diff line number Diff line change
Expand Up @@ -133,3 +133,8 @@ jobs:
python classify/predict.py --imgsz 32 --weights $b --source ../datasets/mnist2560/test/7/60.png # predict
python classify/predict.py --imgsz 32 --weights $m --source data/images/bus.jpg # predict
python export.py --weights $b --img 64 --imgsz 224 --include torchscript # export
python - <<EOF
import torch
for path in '$m', '$b':
model = torch.hub.load('.', 'custom', path=path, source='local')
EOF

0 comments on commit 8fd337b

Please sign in to comment.