From c8f06948945ca8045f2c884fc8daac073118e1d6 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 10 Jun 2020 19:13:04 -0700 Subject: [PATCH] PyTorch Hub update --- hubconf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hubconf.py b/hubconf.py index 8e06beb..5cc3ded 100644 --- a/hubconf.py +++ b/hubconf.py @@ -5,7 +5,7 @@ Usage: model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, channels=3, classes=80) """ -dependencies = ['torch', 'pyyaml'] +dependencies = ['torch', 'yaml'] import torch from models.yolo import Model