From d0f98c0095d2a8c5b3511d8f515eeeab65ef9c3a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 21 Aug 2020 16:20:45 -0700 Subject: [PATCH] Update yolo.py with model_info(verbose=False) --- models/yolo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/yolo.py b/models/yolo.py index 3aaa422..bba015a 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -168,8 +168,8 @@ class Model(nn.Module): self.info() return self - def info(self): # print model information - model_info(self) + def info(self, verbose=False): # print model information + model_info(self, verbose) def parse_model(d, ch): # model_dict, input_channels(3)