From 6062bb7deec2a38da5d437b47267bc1d7eb4582a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 11 Sep 2020 17:44:11 -0700 Subject: [PATCH] print self.yaml['nc'] override update --- models/yolo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/yolo.py b/models/yolo.py index 1a9a2c6..f9929aa 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -71,7 +71,7 @@ class Model(nn.Module): # Define model if nc and nc != self.yaml['nc']: - print('Overriding %s nc=%g with nc=%g' % (cfg, self.yaml['nc'], nc)) + print('Overriding model.yaml nc=%g with nc=%g' % (self.yaml['nc'], nc)) self.yaml['nc'] = nc # override yaml value self.model, self.save = parse_model(deepcopy(self.yaml), ch=[ch]) # model, savelist, ch_out # print([x.shape for x in self.forward(torch.zeros(1, ch, 64, 64))])