Hubconf.py bug fix (#3007)
This commit is contained in:
1 parent
7c89c829e3
commit
31ee54c28c
2 files changed
+2
-1
No files matched your search
+1
-1
@@ -62,7 +62,7 @@ def create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbos
|
||||
|
||||
def custom(path='path/to/model.pt', autoshape=True, verbose=True):
|
||||
# YOLOv5 custom or local model
|
||||
return create(path, autoshape, verbose)
|
||||
return create(path, autoshape=autoshape, verbose=verbose)
|
||||
|
||||
|
||||
def yolov5s(pretrained=True, channels=3, classes=80, autoshape=True, verbose=True):
|
||||
|
||||
Reference in new issue
Block a user