add verbose to parse_model

fcakyon · Apr 23, 2021 · b7bd90f · b7bd90f
1 parent bbfb0cb
commit b7bd90f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolov5/models/yolo.py b/yolov5/models/yolo.py
@@ -83,7 +83,7 @@ def __init__(self, cfg='yolov5s.yaml', ch=3, nc=None, anchors=None, verbose=1):
         if anchors:
             logger.info(f'Overriding model.yaml anchors with anchors={anchors}')
             self.yaml['anchors'] = round(anchors)  # override yaml value
-        self.model, self.save = parse_model(deepcopy(self.yaml), ch=[ch])  # model, savelist
+        self.model, self.save = parse_model(deepcopy(self.yaml), ch=[ch], verbose=verbose)  # model, savelist
         self.names = [str(i) for i in range(self.yaml['nc'])]  # default names
         # print([x.shape for x in self.forward(torch.zeros(1, ch, 64, 64))])