From d490bdd7d6b4775712ce2028537901ec059682c0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 28 May 2022 22:55:03 +0200 Subject: [PATCH] Add best.pt PyTorch Hub inference to CI (#8024) --- .github/workflows/ci-testing.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml index 5c032e205c1a..5f60746370df 100644 --- 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