PyTorch Hub update

5 years ago · c8f0694894
parent a814720403
commit c8f0694894
1 changed files with 1 additions and 1 deletions
--- 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