Browse Source

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

modifyDataloader
Glenn Jocher GitHub 2 years ago
parent
commit
d490bdd7d6
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      .github/workflows/ci-testing.yml

+ 2
- 0
.github/workflows/ci-testing.yml 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

Loading…
Cancel
Save