Procházet zdrojové kódy

PyTorch Hub update

5.0
Glenn Jocher před 4 roky
rodič
revize
c8f0694894
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. +1
    -1
      hubconf.py

+ 1
- 1
hubconf.py Zobrazit soubor

@@ -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

Načítá se…
Zrušit
Uložit