zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add best.pt PyTorch Hub inference to CI (#8024)

This commit is contained in:
Glenn Jocher 2022-05-28 22:55:03 +02:00 committed by GitHub
parent 7433d3829a
commit d490bdd7d6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -114,4 +114,6 @@ jobs:
import torch import torch
model = torch.hub.load('.', 'custom', path='$model', source='local') model = torch.hub.load('.', 'custom', path='$model', source='local')
print(model('data/images/bus.jpg')) print(model('data/images/bus.jpg'))
model = torch.hub.load('.', 'custom', path='$best', source='local')
print(model('data/images/bus.jpg'))
EOF EOF