Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

print self.yaml['nc'] override update

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-09-11 17:44:11 -07:00
1 parent a62a45b2dd
commit 6062bb7dee
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
models/yolo.py
+1 -1
View File
@@ -71,7 +71,7 @@ class Model(nn.Module):
# Define model # Define model
if nc and nc != self.yaml['nc']: 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.yaml['nc'] = nc # override yaml value
self.model, self.save = parse_model(deepcopy(self.yaml), ch=[ch]) # model, savelist, ch_out 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))]) # print([x.shape for x in self.forward(torch.zeros(1, ch, 64, 64))])