PyTorch Hub and autoShape update (#1415)
This commit is contained in:
parent
93fc607f07
commit
7816c0c2d3
|
|
@ -14,7 +14,7 @@ from models.yolo import Model
|
||||||
from utils.general import set_logging
|
from utils.general import set_logging
|
||||||
from utils.google_utils import attempt_download
|
from utils.google_utils import attempt_download
|
||||||
|
|
||||||
dependencies = ['torch', 'PyYAML>=5.3', 'Pillow']
|
dependencies = ['torch', 'yaml', 'Pillow']
|
||||||
set_logging()
|
set_logging()
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue