Spaces:

xiang-wuu
/

yolov5

Runtime error

glenn-jocher commited on May 28, 2022

Commit

d490bdd

•

1 Parent(s): 7433d38

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

Files changed (1) hide show

.github/workflows/ci-testing.yml CHANGED Viewed

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

           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