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:
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
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论