Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

PyTorch Hub update

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
hubconf.py
+1 -1
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