diff --git a/hubconf.py b/hubconf.py index 8455ae2..29e93bd 100644 --- a/hubconf.py +++ b/hubconf.py @@ -27,8 +27,8 @@ def create(name, pretrained, channels, classes): Returns: pytorch model """ + config = os.path.join(os.path.dirname(__file__), 'models', '%s.yaml' % name) # model.yaml path try: - config = os.path.join(os.path.dirname(__file__), 'models', '%s.yaml' % name) # model.yaml path model = Model(config, channels, classes) if pretrained: ckpt = '%s.pt' % name # checkpoint filename @@ -39,7 +39,7 @@ def create(name, pretrained, channels, classes): return model except Exception as e: help_url = 'https://github.com/ultralytics/yolov5/issues/36' - print('%s\nCache is out of date. Delete cache and retry. See %s for help.' % (e, help_url)) + print('%s\nCache maybe be out of date. Delete cache and retry. See %s for help.' % (e, help_url)) def yolov5s(pretrained=False, channels=3, classes=80):