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

ctjanuhowski · Sep 8, 2022 · 4cabe18 · 4cabe18
1 parent 42867f8
commit 4cabe18
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