PyTorch Hub and autoShape update (#1415)

2020-11-16 23:15:43 +01:00 · 2020-11-16 23:15:43 +01:00 · 7816c0c2d3
parent 93fc607f07
commit 7816c0c2d3
1 changed files with 1 additions and 1 deletions
--- a/hubconf.py
+++ b/hubconf.py
@ -14,7 +14,7 @@ from models.yolo import Model
 from utils.general import set_logging
 from utils.google_utils import attempt_download

-dependencies = ['torch', 'PyYAML>=5.3', 'Pillow']
+dependencies = ['torch', 'yaml', 'Pillow']
 set_logging()