Skip to content

Commit 15b7565

Browse files
authored
Update hubconf.py local repo Usage example (#9803)
* Update hubconf.py Signed-off-by: Glenn Jocher <[email protected]> * Update hubconf.py Signed-off-by: Glenn Jocher <[email protected]> Signed-off-by: Glenn Jocher <[email protected]>
1 parent 16f87bb commit 15b7565

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

hubconf.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,10 @@
44
55
Usage:
66
import torch
7-
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
8-
model = torch.hub.load('ultralytics/yolov5:master', 'custom', 'path/to/yolov5s.onnx') # custom model from branch
7+
model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # official model
8+
model = torch.hub.load('ultralytics/yolov5:master', 'yolov5s') # from branch
9+
model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s.pt') # custom/local model
10+
model = torch.hub.load('.', 'custom', 'yolov5s.pt', source='local') # local repo
911
"""
1012

1113
import torch

0 commit comments

Comments
 (0)