PyTorch Hub and autoShape update (ultralytics#1415)

fhkiel-mlaip · Nov 16, 2020 · 9f3e4da · 9f3e4da
1 parent cf6d67e
commit 9f3e4da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hubconf.py b/hubconf.py
@@ -14,7 +14,7 @@
 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()