Print hubconf.py args (ultralytics#8800)

ctjanuhowski · Sep 8, 2022 · 162f652 · 162f652
1 parent 22fbdf9
commit 162f652
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/hubconf.py b/hubconf.py
@@ -128,12 +128,13 @@ def yolov5x6(pretrained=True, channels=3, classes=80, autoshape=True, _verbose=T
     import numpy as np
     from PIL import Image
 
-    from utils.general import cv2
+    from utils.general import cv2, print_args
 
     # Argparser
     parser = argparse.ArgumentParser()
     parser.add_argument('--model', type=str, default='yolov5s', help='model name')
     opt = parser.parse_args()
+    print_args(vars(opt))
 
     # Model
     model = _create(name=opt.model, pretrained=True, channels=3, classes=80, autoshape=True, verbose=True)