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

tdhooghe · Jun 10, 2022 · 9ca710d · 9ca710d
1 parent 1104f83
commit 9ca710d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
@@ -114,4 +114,6 @@ jobs:
           import torch
           model = torch.hub.load('.', 'custom', path='$model', source='local')
           print(model('data/images/bus.jpg'))
+          model = torch.hub.load('.', 'custom', path='$best', source='local')
+          print(model('data/images/bus.jpg'))
           EOF