From e92245a38c90be884a05cb1e3fe325037b9f1379 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 12 Dec 2020 11:42:27 -0800 Subject: [PATCH] Simplified PyTorch hub for custom models (#1677) --- hubconf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hubconf.py b/hubconf.py index 5cd7737..8474fdc 100644 --- a/hubconf.py +++ b/hubconf.py @@ -129,7 +129,7 @@ def custom(path_or_model='path/to/model.pt'): if __name__ == '__main__': model = create(name='yolov5s', pretrained=True, channels=3, classes=80) # pretrained example - # model = custom(model='path/to/model.pt') # custom example + # model = custom(path_or_model='path/to/model.pt') # custom example model = model.autoshape() # for PIL/cv2/np inputs and NMS # Verify inference