zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

PyTorch Hub update

This commit is contained in:
Glenn Jocher 2020-06-10 19:13:04 -07:00
parent a814720403
commit c8f0694894
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
hubconf.py
View File

@ -5,7 +5,7 @@ Usage:
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, channels=3, classes=80) model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, channels=3, classes=80)
""" """
dependencies = ['torch', 'pyyaml'] dependencies = ['torch', 'yaml']
import torch import torch
from models.yolo import Model from models.yolo import Model