Skip to content

Commit 162f652

Browse files
glenn-jocherClay Januhowski
authored andcommitted
Print hubconf.py args (ultralytics#8800)
1 parent 22fbdf9 commit 162f652

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

hubconf.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,12 +128,13 @@ def yolov5x6(pretrained=True, channels=3, classes=80, autoshape=True, _verbose=T
128128
import numpy as np
129129
from PIL import Image
130130

131-
from utils.general import cv2
131+
from utils.general import cv2, print_args
132132

133133
# Argparser
134134
parser = argparse.ArgumentParser()
135135
parser.add_argument('--model', type=str, default='yolov5s', help='model name')
136136
opt = parser.parse_args()
137+
print_args(vars(opt))
137138

138139
# Model
139140
model = _create(name=opt.model, pretrained=True, channels=3, classes=80, autoshape=True, verbose=True)

0 commit comments

Comments
 (0)