From 56f5cb5a28ac8fb5afc49392633763203f37e9bb Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 30 Jul 2022 21:02:26 +0200 Subject: [PATCH] Print hubconf.py args (#8800) --- hubconf.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/hubconf.py b/hubconf.py index f579c64..08122ea 100644 --- a/hubconf.py +++ b/hubconf.py @@ -128,12 +128,13 @@ if __name__ == '__main__': 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)