PyTorch Hub update

This commit is contained in:
Glenn Jocher committed 2020-06-10 19:13:04 -07:00
1 parent a814720403
commit c8f0694894
1 file changed
+1 -1
+1 -1
View File
@@ -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