glenn-jocher commited on
Commit
e92245a
·
unverified ·
1 Parent(s): f7a923b

Simplified PyTorch hub for custom models (#1677)

Browse files
Files changed (1) hide show
  1. hubconf.py +1 -1
hubconf.py CHANGED
@@ -129,7 +129,7 @@ def custom(path_or_model='path/to/model.pt'):
129
 
130
  if __name__ == '__main__':
131
  model = create(name='yolov5s', pretrained=True, channels=3, classes=80) # pretrained example
132
- # model = custom(model='path/to/model.pt') # custom example
133
  model = model.autoshape() # for PIL/cv2/np inputs and NMS
134
 
135
  # Verify inference
 
129
 
130
  if __name__ == '__main__':
131
  model = create(name='yolov5s', pretrained=True, channels=3, classes=80) # pretrained example
132
+ # model = custom(path_or_model='path/to/model.pt') # custom example
133
  model = model.autoshape() # for PIL/cv2/np inputs and NMS
134
 
135
  # Verify inference