@@ -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 |