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

This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-05-28 22:55:03 +02:00
1 parent 7433d3829a
commit d490bdd7d6
1 file changed
+2
+2
View File
@@ -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