From a5818b280edcc983cb6bec626f36db50a5221f68 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 11 Jul 2020 15:04:53 -0700 Subject: [PATCH] expired pytorch hub cache bug fix #365 --- hubconf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hubconf.py b/hubconf.py index 8455ae23f..29e93bdf2 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):