Unverified 提交 d490bdd7 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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

上级 7433d382
...@@ -114,4 +114,6 @@ jobs: ...@@ -114,4 +114,6 @@ jobs:
import torch import torch
model = torch.hub.load('.', 'custom', path='$model', source='local') model = torch.hub.load('.', 'custom', path='$model', source='local')
print(model('data/images/bus.jpg')) print(model('data/images/bus.jpg'))
model = torch.hub.load('.', 'custom', path='$best', source='local')
print(model('data/images/bus.jpg'))
EOF EOF
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论